結果

問題 No.1017 Reiwa Sequence
ユーザー sugarrrsugarrr
提出日時 2020-04-03 22:35:59
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
WA  
実行時間 -
コード長 4,216 bytes
コンパイル時間 2,391 ms
コンパイル使用メモリ 211,808 KB
実行使用メモリ 400,128 KB
最終ジャッジ日時 2024-07-03 04:48:47
合計ジャッジ時間 48,396 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge3
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 277 ms
355,072 KB
testcase_01 AC 267 ms
354,904 KB
testcase_02 AC 273 ms
355,012 KB
testcase_03 AC 270 ms
354,944 KB
testcase_04 AC 270 ms
355,072 KB
testcase_05 AC 281 ms
354,944 KB
testcase_06 AC 275 ms
354,944 KB
testcase_07 AC 275 ms
354,944 KB
testcase_08 AC 269 ms
354,816 KB
testcase_09 AC 273 ms
354,944 KB
testcase_10 AC 270 ms
354,944 KB
testcase_11 AC 272 ms
354,944 KB
testcase_12 AC 270 ms
355,840 KB
testcase_13 AC 268 ms
355,216 KB
testcase_14 AC 268 ms
355,168 KB
testcase_15 AC 272 ms
355,200 KB
testcase_16 AC 285 ms
354,944 KB
testcase_17 AC 276 ms
355,840 KB
testcase_18 AC 270 ms
354,944 KB
testcase_19 AC 512 ms
399,872 KB
testcase_20 AC 471 ms
400,128 KB
testcase_21 AC 478 ms
400,000 KB
testcase_22 AC 444 ms
400,120 KB
testcase_23 AC 445 ms
400,000 KB
testcase_24 AC 478 ms
400,000 KB
testcase_25 AC 448 ms
400,128 KB
testcase_26 AC 436 ms
400,128 KB
testcase_27 AC 301 ms
363,776 KB
testcase_28 AC 317 ms
366,080 KB
testcase_29 AC 293 ms
360,576 KB
testcase_30 AC 275 ms
357,180 KB
testcase_31 AC 299 ms
360,192 KB
testcase_32 AC 287 ms
359,808 KB
testcase_33 AC 269 ms
355,456 KB
testcase_34 AC 265 ms
354,816 KB
testcase_35 AC 256 ms
354,816 KB
testcase_36 AC 260 ms
354,936 KB
testcase_37 AC 257 ms
354,944 KB
testcase_38 AC 356 ms
376,448 KB
testcase_39 AC 271 ms
357,376 KB
testcase_40 WA -
testcase_41 WA -
testcase_42 WA -
testcase_43 WA -
testcase_44 WA -
testcase_45 WA -
testcase_46 WA -
testcase_47 WA -
testcase_48 AC 287 ms
358,528 KB
testcase_49 WA -
testcase_50 AC 275 ms
358,468 KB
testcase_51 AC 261 ms
355,012 KB
testcase_52 AC 297 ms
365,312 KB
testcase_53 AC 476 ms
399,488 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
//#include "bits/stdc++.h"
using namespace std;
typedef long long ll;
//#include "boost/multiprecision/cpp_int.hpp"
//typedef boost::multiprecision::cpp_int LL;
typedef long double dd;
#define i_7 (ll)(1E9+7)
//#define i_7 998244353
#define i_5 i_7-2

ll mod(ll a){
    ll c=a%i_7;
    if(c>=0)return c;
    return c+i_7;
}
typedef pair<ll,ll> l_l;
typedef pair<dd,dd> d_d;
ll inf=(ll)1E16;
#define rep(i,l,r) for(ll i=l;i<=r;i++)
#define pb push_back
ll max(ll a,ll b){if(a<b)return b;else return a;}
ll min(ll a,ll b){if(a>b)return b;else return a;}
void Max(ll &pos,ll val){pos=max(pos,val);}//Max(dp[n],dp[n-1]);
void Min(ll &pos,ll val){pos=min(pos,val);}
void Add(ll &pos,ll val){pos=mod(pos+val);}
dd EPS=1E-9;
#define fastio ios::sync_with_stdio(false); cin.tie(0); cout.tie(0);
#define fi first
#define se second
#define endl "\n"
#define SORT(v) sort(v.begin(),v.end())
#define ERASE(v) v.erase(unique(v.begin(),v.end()),v.end())
#define POSL(v,x) (lower_bound(v.begin(),v.end(),x)-v.begin())
#define POSU(v,x) (upper_bound(v.begin(),v.end(),x)-v.begin())
//template<class T>void max(T a,T b){if(a<b)return b;else return a;}
//template<class T>void min(T a,T b){if(a>b)return b;else return a;}
//template<class T>bool Max(T&a, T b){if(a < b){a = b;return 1;}return 0;}
//template<class T>bool Min(T&a, T b){if(a > b){a = b;return 1;}return 0;}

//////////////////////////

#define M 15000005
vector<ll>fin[M];

vector<ll>X,Y;
ll ori[150005];
ll n;
void dfs(ll pos,vector<ll>v,ll sum){
    if(pos==n)return;
    if(X.size()>=1)return;
    dfs(pos+1,v,sum);
    if(X.size()>=1)return;
    v.pb(pos);
    ll newsum=sum+ori[pos];
    if(fin[newsum].size()>=1){
        X=v;
        Y=fin[newsum];
        return;
    }
    fin[newsum]=v;
    dfs(pos+1,v,newsum);
    if(X.size()>=1)return;
    v.pop_back();
}

int main(){fastio
    cin>>n;
    
    l_l a[n];
    rep(i,0,n-1){cin>>a[i].fi;a[i].se=i;ori[i]=a[i].fi;}
    sort(a,a+n);
    rep(i,0,n-2){
        if(a[i].fi==a[i+1].fi){
            ll x=a[i].se,y=a[i+1].se;
            cout<<"Yes"<<endl;
            rep(pos,0,n-1){
                if(pos==x)cout<<ori[pos]<<" ";
                else if(pos==y)cout<<-ori[pos]<<" ";
                else cout<<"0 ";
            }
            cout<<endl;
            return 0;
        }
    }
    
    if(n>=1000){
        ll N=300005;
        l_l dp[N];rep(i,0,N-1)dp[i]={-1,-1};
        rep(i,0,n-1){
            rep(j,i+1,n-1){
                ll c=ori[i]+ori[j];
                if(dp[c].fi!=-1){
                    cout<<"Yes"<<endl;
                    rep(pos,0,n-1){
                        if(pos==i||pos==j)cout<<ori[pos]<<" ";
                        else if(pos==dp[c].fi||pos==dp[c].se)cout<<ori[pos]<<" ";
                        else cout<<"0 ";
                    }
                    cout<<endl;return 0;
                }
                dp[c]={i,j};
            }
        }
    }
    
    if(n>=100){
        ll ans[n];memset(ans,0,sizeof(ans));
        ll N=600005;
        vector<ll>dp[N];
        rep(i,0,n-1){
            rep(j,i+1,n-1){
                rep(k,j+1,n-1){
                    rep(l,k+1,n-1){
                        ll c=ori[i]+ori[j]+ori[k]+ori[l];
                        if(dp[c].size()>=1){
                            cout<<"Yes"<<endl;
                            for(auto x:dp[c]){
                                ans[x]++;
                            }
                            ans[i]--;ans[j]--;ans[k]--;ans[l]--;
                            
                            rep(pos,0,n-1){
                                cout<<ori[pos]*ans[pos]<<" ";
                            }cout<<endl;
                            return 0;
                        }
                        dp[c].pb(i);dp[c].pb(j);dp[c].pb(k);dp[c].pb(l);
                    }
                }
            }
        }
    }
    vector<ll>v;
    dfs(0,v,0);
    ll ans[n];memset(ans,0,sizeof(ans));

    if(X.size()>=1){
        cout<<"Yes"<<endl;
        for(auto x:X)ans[x]++;
        for(auto y:Y)ans[y]--;
        rep(pos,0,n-1){
            cout<<ori[pos]*ans[pos]<<" ";
        }cout<<endl;
        return 0;
    }
    cout<<"No"<<endl;
    
    return 0;
}
0