結果

問題 No.2982 Logic Battle
ユーザー askr58
提出日時 2024-12-07 02:12:59
言語 C++23
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 1,632 ms / 2,000 ms
コード長 1,602 bytes
コンパイル時間 5,877 ms
コンパイル使用メモリ 315,076 KB
実行使用メモリ 5,248 KB
最終ジャッジ日時 2024-12-07 02:13:43
合計ジャッジ時間 40,017 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 38
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
#include <atcoder/all>
using namespace std;
using namespace atcoder;
using ll=long long;
using mint=modint998244353;
int main()
{
    int n;
    cin>>n;
    ll inf=1000000000000000000;
    vector<vector<ll>> a(n,vector<ll>(3));
    for(int i=0;i<n;i++)cin>>a[i][0]>>a[i][1]>>a[i][2];
    vector<vector<ll>> crt(n+1,vector<ll>(3,-inf));
    for(int j=0;j<3;j++){
        if(a[0][j]>=n){
            crt[n][j]=max(crt[n][j],a[0][j]*n-n*(n-1)/2);
        }else if(a[0][j]>0){
            crt[a[0][j]-1][j]=max(crt[a[0][j]-1][j],a[0][j]);
        }else{
            crt[0][j]=0;
        }
    }
    for(int i=1;i<n;i++){
        vector<vector<ll>> nxt(n+1,vector<ll>(3,-inf));
        for(int j=0;j<=n;j++){
            for(int k=0;k<3;k++){
                if(crt[j][k]==-inf)continue;
                for(int l=0;l<3;l++){
                    if(k!=l){
                        if(j==n){
                            nxt[n][l]=max(nxt[n][l],crt[j][k]+a[i][l]*(n-i));
                        }else if(j+a[i][l]>=n){
                            nxt[n][l]=max(nxt[n][l],crt[j][k]+(j+a[i][l])*(n-i)-(n-i)*(n-i-1)/2);
                        }else if(j+a[i][l]>0){
                            nxt[j+a[i][l]-1][l]=max(nxt[j+a[i][l]-1][l],crt[j][k]+j+a[i][l]);
                        }else{
                            nxt[0][l]=max(nxt[0][l],crt[j][k]);
                        }
                    }
                }
            }
        }
        crt=nxt;
    }
    ll ans=0;
    for(int i=0;i<=n;i++)for(int j=0;j<3;j++)ans=max(ans,crt[i][j]);
    cout<<ans<<endl;
    return 0;
}
0