結果
問題 | No.1017 Reiwa Sequence |
ユーザー | Rubikun |
提出日時 | 2020-04-03 23:18:43 |
言語 | C++14 (gcc 12.3.0 + boost 1.83.0) |
結果 |
TLE
|
実行時間 | - |
コード長 | 8,136 bytes |
コンパイル時間 | 2,128 ms |
コンパイル使用メモリ | 190,068 KB |
実行使用メモリ | 30,208 KB |
最終ジャッジ日時 | 2024-07-03 06:14:01 |
合計ジャッジ時間 | 17,947 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge5 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 3 ms
10,624 KB |
testcase_01 | AC | 3 ms
5,376 KB |
testcase_02 | AC | 2 ms
5,376 KB |
testcase_03 | AC | 4 ms
5,504 KB |
testcase_04 | AC | 2 ms
5,376 KB |
testcase_05 | AC | 4 ms
5,504 KB |
testcase_06 | AC | 4 ms
5,632 KB |
testcase_07 | AC | 3 ms
5,376 KB |
testcase_08 | AC | 2 ms
5,376 KB |
testcase_09 | AC | 14 ms
5,504 KB |
testcase_10 | AC | 35 ms
5,760 KB |
testcase_11 | AC | 7 ms
5,504 KB |
testcase_12 | AC | 247 ms
30,208 KB |
testcase_13 | AC | 61 ms
13,184 KB |
testcase_14 | AC | 13 ms
5,376 KB |
testcase_15 | AC | 35 ms
5,376 KB |
testcase_16 | AC | 14 ms
5,376 KB |
testcase_17 | AC | 303 ms
5,376 KB |
testcase_18 | AC | 13 ms
5,376 KB |
testcase_19 | TLE | - |
testcase_20 | -- | - |
testcase_21 | -- | - |
testcase_22 | -- | - |
testcase_23 | -- | - |
testcase_24 | -- | - |
testcase_25 | -- | - |
testcase_26 | -- | - |
testcase_27 | -- | - |
testcase_28 | -- | - |
testcase_29 | -- | - |
testcase_30 | -- | - |
testcase_31 | -- | - |
testcase_32 | -- | - |
testcase_33 | -- | - |
testcase_34 | -- | - |
testcase_35 | -- | - |
testcase_36 | -- | - |
testcase_37 | -- | - |
testcase_38 | -- | - |
testcase_39 | -- | - |
testcase_40 | -- | - |
testcase_41 | -- | - |
testcase_42 | -- | - |
testcase_43 | -- | - |
testcase_44 | -- | - |
testcase_45 | -- | - |
testcase_46 | -- | - |
testcase_47 | -- | - |
testcase_48 | -- | - |
testcase_49 | -- | - |
testcase_50 | -- | - |
testcase_51 | -- | - |
testcase_52 | -- | - |
testcase_53 | -- | - |
ソースコード
#include <bits/stdc++.h> using namespace std; typedef long long ll; #define all(x) (x).begin(),(x).end() const int mod=1000000007,MAX=200005,INF=1<<30; int main(){ std::ifstream in("text.txt"); std::cin.rdbuf(in.rdbuf()); cin.tie(0); ios::sync_with_stdio(false); int N;cin>>N; vector<int> A(N),cnt(150003),B; for(int i=0;i<N;i++){ cin>>A[i]; cnt[A[i]]++; } B=A; for(int i=0;i<150003;i++){ if(cnt[i]>=2){ cout<<"Yes"<<endl; int a=0; for(int j=0;j<N;j++){ if(A[j]==i){ if(a==0) a++; else if(a==1){ A[j]*=-1; a++; }else A[j]=0; }else A[j]=0; } for(int j=0;j<N;j++){ if(j) cout<<" "; cout<<A[j]; } cout<<endl; return 0; } } sort(all(A)); if(N>=1000){ vector<pair<int,int>> when(150003,{-1,-1}); for(int i=0;i+1<N;i++){ int a=when[A[i+1]-A[i]].first; if(a!=-1&&a!=i-1){ int p1=A[when[A[i+1]-A[i]].first],p2=A[i+1]; int m1=A[when[A[i+1]-A[i]].second],m2=A[i]; for(int j=0;j<N;j++){ if(B[j]==p1||B[j]==p2) continue; if(B[j]==m1||B[j]==m2) B[j]*=-1; else B[j]=0; } cout<<"Yes"<<endl; for(int j=0;j<N;j++){ if(j) cout<<" "; cout<<B[j]; } cout<<endl; return 0; } if(a==-1){ when[A[i+1]-A[i]]={i,i+1}; } } } vector<pair<int,int>> when(150003,{-1,-1}); for(int l=1;l<N;l++){ for(int i=0;i+l<N;i++){ int j=i+l; int a=when[A[j]-A[i]].first,b=when[A[j]-A[i]].second; if(i==a||i==b||j==a||j==b) continue; if(a!=-1){ for(int k=0;k<N;k++){ if(B[k]==A[i]||B[k]==A[b]) continue; if(B[k]==A[j]||B[k]==A[a]) B[k]*=-1; else B[k]=0; } cout<<"Yes"<<endl; for(int k=0;k<N;k++){ if(k) cout<<" "; cout<<B[k]; } cout<<endl; return 0; } if(a==-1){ when[A[j]-A[i]]={i,j}; } } } if(N<=20){ for(int bit=0;bit<(1<<N);bit++){ for(int U=bit;;U=(U-1)&bit){ int V=bit-U; ll suma=0,sumb=0; for(int i=0;i<N;i++){ if(U&(1<<i)) suma+=A[i]; if(V&(1<<i)) sumb+=A[i]; } if(suma&&suma==sumb){ vector<int> use(150003); for(int i=0;i<N;i++){ if(U&(1<<i)) use[A[i]]=1; if(V&(1<<i)) use[A[i]]=2; } for(int i=0;i<N;i++){ if(use[B[i]]==0) B[i]=0; else if(use[B[i]]==2) B[i]*=-1; } cout<<"Yes"<<endl; for(int i=0;i<N;i++){ if(i) cout<<" "; cout<<B[i]; } cout<<endl; return 0; } if(U==0) break; } } cout<<"No"<<endl; return 0; } A=B; map<ll,pair<vector<int>,vector<int>>> MA; for(int bit=0;bit<(1<<(N/2));bit++){ for(int U=bit;;U=(U-1)&bit){ int V=bit-U; vector<int> a,b; ll suma=0,sumb=0; for(int i=0;i<N/2;i++){ if(U&(1<<i)){ suma+=A[i]; a.push_back(i); } if(V&(1<<i)){ sumb+=A[i]; b.push_back(i); } } if(suma&&sumb){ MA[suma-sumb]={a,b}; if(suma==sumb){ for(int i=0;i<N;i++){ bool ok=false; for(int j=0;j<a.size();j++){ if(i==a[j]) ok=true; } for(int j=0;j<b.size();j++){ if(i==b[j]){ A[i]*=-1; ok=true; } } if(!ok) A[i]=0; } cout<<"Yes"<<endl; for(int i=0;i<N;i++){ if(i) cout<<" "; cout<<A[i]; } cout<<endl; return 0; } } if(U==0) break; } } for(int bit=0;bit<(1<<(N-N/2));bit++){ for(int U=bit;;U=(U-1)&bit){ int V=bit-U; vector<int> a,b; ll suma=0,sumb=0; for(int i=N/2;i<N;i++){ if(U&(1<<(i-N/2))){ suma+=A[i]; a.push_back(i); } if(V&(1<<(i-N/2))){ sumb+=A[i]; b.push_back(i); } } if(suma&&sumb){ if(MA.count(sumb-suma)){ vector<int> c=MA[sumb-suma].first,d=MA[sumb-suma].second; for(int x:c) a.push_back(x); for(int x:d) b.push_back(x); for(int i=0;i<N;i++){ bool ok=false; for(int j=0;j<a.size();j++){ if(i==a[j]) ok=true; } for(int j=0;j<b.size();j++){ if(i==b[j]){ A[i]*=-1; ok=true; } } if(!ok) A[i]=0; } cout<<"Yes"<<endl; for(int i=0;i<N;i++){ if(i) cout<<" "; cout<<A[i]; } cout<<endl; return 0; } if(suma==sumb){ for(int i=0;i<N;i++){ bool ok=false; for(int j=0;j<a.size();j++){ if(i==a[j]) ok=true; } for(int j=0;j<b.size();j++){ if(i==b[j]){ A[i]*=-1; ok=true; } } if(!ok) A[i]=0; } cout<<"Yes"<<endl; for(int i=0;i<N;i++){ if(i) cout<<" "; cout<<A[i]; } cout<<endl; return 0; } } if(U==0) break; } } cout<<"No"<<endl; }