結果

問題 No.1017 Reiwa Sequence
ユーザー sugarrr
提出日時 2020-04-03 22:37:25
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 435 ms / 2,000 ms
コード長 4,217 bytes
コンパイル時間 2,384 ms
コンパイル使用メモリ 206,200 KB
最終ジャッジ日時 2025-01-09 13:21:50
ジャッジサーバーID
（参考情報） 		judge2 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 50
権限があれば一括ダウンロードができます

ソースコード

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