結果

問題 No.5021 Addition Pyramid
ユーザー FplusFplusF
提出日時 2025-02-25 22:57:20
言語 C++23
(gcc 13.3.0 + boost 1.87.0)
結果
TLE  
実行時間 -
コード長 5,077 bytes
コンパイル時間 4,690 ms
コンパイル使用メモリ 319,820 KB
実行使用メモリ 6,824 KB
スコア 229,782,475
最終ジャッジ日時 2025-02-25 22:59:06
合計ジャッジ時間 100,263 ms
ジャッジサーバーID
（参考情報） 		judge7 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 48 TLE * 2
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;
using pii=pair<int,int>;
using tii=tuple<int,int,int>;
using qii=tuple<int,int,int,int>;
using ll=long long;
using ull=unsigned long long;
using ld=long double;
constexpr int INF=1e9;
constexpr ll INF_ll=1e18;
#define rep(i,n) for (int i=0;i<(int)(n);i++)
#define replr(i,l,r) for (int i=(int)(l);i<(int)(r);i++)
#define all(v) v.begin(),v.end()
#define len(v) ((int)v.size())
template<class T> inline bool chmin(T &a,T b){
    if(a>b){
        a=b;
        return true;
    }
    return false;
}
template<class T> inline bool chmax(T &a,T b){
    if(a<b){
        a=b;
        return true;
    }
    return false;
}

namespace Timer{
    auto program_start=chrono::system_clock::now();
    auto start=program_start;
    void snap(){
        start=chrono::system_clock::now();
    }
    int get_ms(){
        auto now=chrono::system_clock::now();
        int ms=chrono::duration_cast<chrono::milliseconds>(now-start).count();
        return ms;
    }
    int get_ms_all_program(){
        auto now=chrono::system_clock::now();
        int ms=chrono::duration_cast<chrono::milliseconds>(now-program_start).count();
        return ms;
    }
}

mt19937 mt;
uint32_t rand_int(uint32_t r){  //[0,r)
    assert(r!=0);
    return ((uint64_t)mt()*r)>>32;
}
int rand_int(int l,int r){  //[l,r)
    assert(l<r);
    return l+rand_int(r-l);
}
constexpr double one_div_mt_max=1.0/(double)mt19937::max();
double rand_double(){  //[0.0,1.0]
    return mt()*one_div_mt_max;
}

constexpr int N=50,mod=100000000;
vector<vector<int>> A;

#include <atcoder/modint>
using mint=atcoder::static_modint<mod>;

vector<vector<vector<mint>>> C;

namespace Solver{
    vector<vector<pii>> now_calc;
    struct State{
        vector<mint> ans;
        int score=0;
        int idx=0;

        void update_score(int max_score,int pos){
            for(auto [i,j]:now_calc[pos]){
                mint now=0;
                rep(k,N){
                    now+=ans[k]*C[i][j][k];
                }
                int v=min(abs<int>(A[i][j]-now.val()),mod-abs<int>(A[i][j]-now.val()));
                chmax(score,v);
                if(max_score<=score) return;
            }
        }

        State():ans(N,0),idx(0){}
        void advance(int v,int max_score,int pos){
            ans[pos]=v;
            update_score(max_score,pos);
            idx++;
        }
        auto operator<=>(const State &other) const{
            return this->score<=>other.score;
        }
    };
    constexpr int K=250,W=250;
    void solve(){

        vector<mint> last_ans;

        vector<int> ord(N);
        rep(i,N) ord[i]=i;
        deque<int> dq(all(ord));

        vector<int> new_ord;


        int min_score=INF;

        while(!dq.empty()){
            new_ord.push_back(dq.front());
            dq.pop_front();
            new_ord.push_back(dq.back());
            dq.pop_back();
        }
        reverse(all(new_ord));
        ord=new_ord;

        vector<int> t(N,0);
        rep(i,N) t[ord[i]]=i;



        now_calc.assign(N,{});
        rep(i,N){
            rep(j,i+1){
                int mx=-1,idx=-1;
                rep(k,N){
                    if(C[i][j][k]==0) continue;
                    if(chmax(mx,t[k])) idx=k;
                }
                now_calc[idx].emplace_back(i,j);
            }
        }

        priority_queue<State> beam;
        beam.push(State());
        rep(d,N){

            
            vector<State> beam_v;
            
            while(!beam.empty()){
                beam_v.push_back(beam.top());
                beam.pop();
            }
            reverse(all(beam_v));
            
            priority_queue<State> new_beam;
            rep(w,len(beam_v)){
                const auto &state=beam_v[w];
                rep(i,K){
                    int max_score=INF;
                    if(W<=len(new_beam)) max_score=new_beam.top().score;
                    State new_state=state;
                    new_state.advance(mod/K*i,max_score,ord[d]);
                    if(max_score<=new_state.score) continue;
                    new_beam.push(new_state);
                    while(W<len(new_beam)) new_beam.pop();
                }
            }
            swap(beam,new_beam);
        }
        
        State state=beam.top();
        if(chmin(min_score,state.score)) last_ans=state.ans;
        rep(i,N){
            cout << last_ans[i].val() << ' ';
        }
        cout << '\n';
        cerr << mod/2-min_score << ' ' << Timer::get_ms() << '\n';
    }
}

int main(){
    ios::sync_with_stdio(false);
    cin.tie(nullptr);
    Timer::snap();
    int _;
    cin >> _;
    A.resize(N);
    rep(i,N){
        A[i].resize(i+1);
        rep(j,i+1){
            cin >> A[i][j];
        }
    }

    C.assign(N,vector<vector<mint>>(N,vector<mint>(N,0)));

    rep(i,N) C[N-1][i][i]=1;

    for(int i=N-2;0<=i;i--){
        rep(j,i+1){
            rep(k,N){
                C[i][j][k]+=C[i+1][j][k];
                C[i][j][k]+=C[i+1][j+1][k];
            }
        }
    }

    Solver::solve();
    exit(0);
}
0