結果

問題 No.5021 Addition Pyramid
ユーザー FplusFplusF
提出日時 2025-02-25 20:24:59
言語 C++23
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 1,903 ms / 2,000 ms
コード長 3,395 bytes
コンパイル時間 4,951 ms
コンパイル使用メモリ 295,324 KB
実行使用メモリ 6,824 KB
スコア 1,846,195
最終ジャッジ日時 2025-02-25 20:26:44
合計ジャッジ時間 103,497 ms
ジャッジサーバーID
(参考情報)
judge5 / 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=10,mod=100000000;
vector<vector<int>> A;

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

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

namespace Solver{
    double calc_score(vector<mint> ans){
        double ret=0;
        rep(i,N){
            rep(j,i+1){
                mint now=0;
                rep(k,N){
                    now+=ans[k]*C[i][j][k];
                }
                int v=now.val();
                ret+=pow(min(abs(A[i][j]-v),mod-abs(A[i][j]-v)),2);
            }
        }
        return ret;
    }
    void solve(){
        vector<mint> ans(N,0);
        double pre_score=calc_score(ans);
        while(Timer::get_ms()<1900){
            int x=rand_int(N);
            vector<mint> new_ans=ans;
            new_ans[x]=rand_int(mod);
            double new_score=calc_score(new_ans);
            if(chmin(pre_score,new_score)) ans=new_ans;
        }
        rep(i,N) cout << ans[i].val() << ' ';
        cout << '\n';

        int score=0;

        rep(i,N){
            rep(j,i+1){
                mint now=0;
                rep(k,N){
                    now+=ans[k]*C[i][j][k];
                }
                int v=now.val();
                chmax(score,min(abs(A[i][j]-v),mod-abs(A[i][j]-v)));
            }
        }
        score=mod/2-score;
        cerr << score << '\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