結果

問題 No.1017 Reiwa Sequence
ユーザー RubikunRubikun
提出日時 2020-04-03 23:17:20
言語 C++14
(gcc 12.3.0 + boost 1.83.0)
結果
WA  
実行時間 -
コード長 6,881 bytes
コンパイル時間 1,947 ms
コンパイル使用メモリ 189,108 KB
実行使用メモリ 30,208 KB
最終ジャッジ日時 2024-07-03 06:11:59
合計ジャッジ時間 32,205 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge1
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 3 ms
5,248 KB
testcase_01 AC 3 ms
5,376 KB
testcase_02 AC 2 ms
5,376 KB
testcase_03 WA -
testcase_04 AC 2 ms
5,376 KB
testcase_05 WA -
testcase_06 WA -
testcase_07 AC 3 ms
5,376 KB
testcase_08 AC 2 ms
5,376 KB
testcase_09 WA -
testcase_10 AC 17 ms
8,064 KB
testcase_11 AC 4 ms
5,376 KB
testcase_12 AC 202 ms
30,208 KB
testcase_13 AC 52 ms
13,312 KB
testcase_14 AC 3 ms
5,376 KB
testcase_15 AC 3 ms
5,376 KB
testcase_16 AC 3 ms
5,376 KB
testcase_17 AC 5 ms
5,376 KB
testcase_18 AC 4 ms
5,376 KB
testcase_19 AC 46 ms
7,808 KB
testcase_20 AC 45 ms
7,808 KB
testcase_21 AC 42 ms
7,168 KB
testcase_22 AC 43 ms
7,680 KB
testcase_23 AC 41 ms
7,424 KB
testcase_24 AC 39 ms
6,912 KB
testcase_25 AC 40 ms
7,424 KB
testcase_26 AC 37 ms
6,784 KB
testcase_27 AC 30 ms
7,936 KB
testcase_28 AC 36 ms
7,680 KB
testcase_29 AC 18 ms
7,936 KB
testcase_30 AC 15 ms
7,040 KB
testcase_31 AC 17 ms
7,936 KB
testcase_32 AC 24 ms
7,424 KB
testcase_33 AC 15 ms
6,528 KB
testcase_34 AC 2 ms
5,376 KB
testcase_35 AC 3 ms
5,376 KB
testcase_36 AC 2 ms
5,376 KB
testcase_37 AC 29 ms
6,272 KB
testcase_38 AC 3 ms
5,376 KB
testcase_39 WA -
testcase_40 AC 26 ms
5,888 KB
testcase_41 AC 26 ms
5,632 KB
testcase_42 AC 25 ms
5,760 KB
testcase_43 AC 25 ms
5,760 KB
testcase_44 AC 21 ms
5,632 KB
testcase_45 AC 25 ms
5,632 KB
testcase_46 AC 25 ms
5,760 KB
testcase_47 AC 23 ms
5,760 KB
testcase_48 AC 28 ms
5,376 KB
testcase_49 AC 28 ms
6,144 KB
testcase_50 AC 26 ms
5,376 KB
testcase_51 AC 4 ms
5,376 KB
testcase_52 AC 32 ms
5,632 KB
testcase_53 AC 23 ms
7,680 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#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};
            }
        }
    }
    
    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;
}
0