結果

問題 No.1017 Reiwa Sequence
ユーザー Shibuyap
提出日時 2020-07-26 02:33:14
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 646 ms / 2,000 ms
コード長 1,417 bytes
コンパイル時間 2,125 ms
コンパイル使用メモリ 204,668 KB
最終ジャッジ日時 2025-01-12 05:45:38
ジャッジサーバーID
（参考情報） 		judge4 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 50
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
#define rep(i,n) for(int i = 0; i < (n); ++i)
#define srep(i,s,t) for (int i = s; i < t; ++i)
#define drep(i,n) for(int i = (n)-1; i >= 0; --i)
using namespace std;
typedef long long int ll;
typedef pair<int,int> P;
#define yn {puts("YES");}else{puts("NO");}
#define MAX_N 200005

int ans[MAX_N];
int a[MAX_N];
int n;
int ok = 0;

map<int,vector<int>> mp;
vector<int> tmp;
int x;

void dfs(int i, int j){
    if(ok)return;
    if(i == n || j >= 22){
        return;
    }
    dfs(i+1,j);
    if(ok)return;
    int y = x + a[i];
    int xx = x;
    if(mp.find(y) != mp.end()){
        ok = 1;
        vector<int> keep = mp[y];
        tmp.push_back(i);

        rep(i,keep.size()){
            // cout << keep[i] << ' ';
            ans[keep[i]]++;
        }
        // cout << endl;

        rep(i,tmp.size()){
            // cout << tmp[i] << ' ';
            ans[tmp[i]]--;
        }
        // cout << endl;

        return;
    }else{
        x = y;
        tmp.push_back(i);
        mp[y] = tmp;
        dfs(i+1,j+1);
    }

    x = xx;
    tmp.erase(tmp.end()-1);

}

int main() {
    cin >> n;
    rep(i,n)cin >> a[i];

    dfs(0,0);

    if(ok == 0){
        cout << "No" << endl;
    }else{
        cout << "Yes" << endl;
        rep(i,n){
            cout << a[i] * ans[i];
            if(i == n - 1) cout << endl;
            else cout << " ";
        }
    }

    return 0;
}
0