結果

問題 No.1017 Reiwa Sequence
ユーザー sugarrrsugarrr
提出日時 2020-04-03 22:37:25
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 500 ms / 2,000 ms
コード長 4,217 bytes
コンパイル時間 2,727 ms
コンパイル使用メモリ 212,380 KB
実行使用メモリ 400,708 KB
最終ジャッジ日時 2024-07-03 04:53:47
合計ジャッジ時間 50,373 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge1
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 220 ms
355,080 KB
testcase_01 AC 214 ms
354,876 KB
testcase_02 AC 219 ms
355,084 KB
testcase_03 AC 224 ms
355,112 KB
testcase_04 AC 214 ms
354,904 KB
testcase_05 AC 153 ms
355,056 KB
testcase_06 AC 222 ms
355,140 KB
testcase_07 AC 220 ms
355,084 KB
testcase_08 AC 171 ms
355,032 KB
testcase_09 AC 151 ms
356,032 KB
testcase_10 AC 221 ms
355,160 KB
testcase_11 AC 217 ms
355,104 KB
testcase_12 AC 154 ms
356,800 KB
testcase_13 AC 150 ms
355,400 KB
testcase_14 AC 147 ms
355,136 KB
testcase_15 AC 227 ms
355,400 KB
testcase_16 AC 151 ms
355,152 KB
testcase_17 AC 149 ms
356,036 KB
testcase_18 AC 142 ms
355,100 KB
testcase_19 AC 500 ms
399,964 KB
testcase_20 AC 365 ms
400,576 KB
testcase_21 AC 378 ms
400,200 KB
testcase_22 AC 361 ms
400,044 KB
testcase_23 AC 349 ms
400,196 KB
testcase_24 AC 395 ms
400,072 KB
testcase_25 AC 350 ms
400,068 KB
testcase_26 AC 347 ms
400,708 KB
testcase_27 AC 187 ms
363,844 KB
testcase_28 AC 193 ms
366,532 KB
testcase_29 AC 168 ms
360,548 KB
testcase_30 AC 154 ms
357,332 KB
testcase_31 AC 173 ms
360,204 KB
testcase_32 AC 249 ms
359,996 KB
testcase_33 AC 171 ms
355,600 KB
testcase_34 AC 141 ms
355,100 KB
testcase_35 AC 141 ms
355,396 KB
testcase_36 AC 144 ms
355,092 KB
testcase_37 AC 149 ms
355,264 KB
testcase_38 AC 247 ms
377,156 KB
testcase_39 AC 239 ms
357,304 KB
testcase_40 AC 239 ms
362,308 KB
testcase_41 AC 236 ms
362,180 KB
testcase_42 AC 235 ms
362,184 KB
testcase_43 AC 235 ms
361,784 KB
testcase_44 AC 233 ms
361,800 KB
testcase_45 AC 233 ms
361,944 KB
testcase_46 AC 237 ms
361,920 KB
testcase_47 AC 236 ms
362,184 KB
testcase_48 AC 247 ms
358,468 KB
testcase_49 AC 240 ms
363,152 KB
testcase_50 AC 175 ms
358,472 KB
testcase_51 AC 142 ms
355,232 KB
testcase_52 AC 187 ms
365,328 KB
testcase_53 AC 499 ms
399,808 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