結果

問題 No.1017 Reiwa Sequence
ユーザー RubikunRubikun
提出日時 2020-04-03 22:45:08
言語 C++14
(gcc 12.3.0 + boost 1.83.0)
結果
TLE  
実行時間 -
コード長 3,783 bytes
コンパイル時間 2,046 ms
コンパイル使用メモリ 178,680 KB
実行使用メモリ 11,932 KB
最終ジャッジ日時 2024-07-03 05:10:52
合計ジャッジ時間 13,344 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge3
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 3 ms
10,752 KB
testcase_01 AC 3 ms
5,376 KB
testcase_02 AC 2 ms
5,376 KB
testcase_03 AC 4 ms
5,632 KB
testcase_04 AC 2 ms
5,376 KB
testcase_05 AC 3 ms
5,632 KB
testcase_06 AC 4 ms
5,632 KB
testcase_07 AC 2 ms
5,376 KB
testcase_08 AC 2 ms
5,376 KB
testcase_09 AC 7 ms
5,504 KB
testcase_10 AC 1,317 ms
5,632 KB
testcase_11 AC 9 ms
5,632 KB
testcase_12 TLE -
testcase_13 -- -
testcase_14 -- -
testcase_15 -- -
testcase_16 -- -
testcase_17 -- -
testcase_18 -- -
testcase_19 -- -
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 -- -
権限があれば一括ダウンロードができます

ソースコード

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 i=0;i<N;i++){
        for(int j=i+1;j<N;j++){
            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};
            }
        }
    }
    
    for(int bit=0;bit<(1<<N);bit++){
        for(int bit2=0;bit2<(1<<N);bit2++){
            if(bit&bit2) continue;
            
            ll suma=0,sumb=0;
            for(int i=0;i<N;i++){
                if(bit&(1<<i)) suma+=A[i];
                if(bit2&(1<<i)) sumb+=A[i];
            }
            
            if(suma&&suma==sumb){
                vector<int> use(150003);
                for(int i=0;i<N;i++){
                    if(bit&(1<<i)) use[A[i]]=1;
                    if(bit2&(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;
            }
        }
    }
    
    cout<<"No"<<endl;
}
0