結果

問題 No.5021 Addition Pyramid
ユーザー FplusFplusF
提出日時 2025-02-25 21:35:11
言語 C++23
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 1,661 ms / 2,000 ms
コード長 4,191 bytes
コンパイル時間 4,475 ms
コンパイル使用メモリ 306,304 KB
実行使用メモリ 6,820 KB
スコア 258,658,783
最終ジャッジ日時 2025-02-25 21:36:40
合計ジャッジ時間 87,114 ms
ジャッジサーバーID
（参考情報） 		judge7 / judge1
純コード判定しない問題か言語
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 50
権限があれば一括ダウンロードができます

ソースコード

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{
    struct State{
        vector<mint> ans;
        int score=0;
        int idx=0;

        void update_score(int max_score){
            int j=0;
            replr(i,N-1-idx,N){
                mint now=0;
                rep(k,idx+1){
                    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;
                j++;
            }
        }

        State():ans(N,0),idx(0){}
        void advance(int v,int max_score){
            ans[idx]=v;
            update_score(max_score);
            idx++;
        }
        auto operator<=>(const State &other) const{
            return this->score<=>other.score;
        }
    };
    constexpr int K=250,W=250;
    void solve(){
        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);
                    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();
        rep(i,N){
            cout << state.ans[i].val() << ' ';
        }
        cout << '\n';
        cerr << mod/2-state.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