結果

問題 No.2037 NAND Pyramid
ユーザー keisuke6
提出日時 2022-08-12 23:18:46
言語 C++23
(gcc 13.3.0 + boost 1.87.0)
結果
TLE  
実行時間 -
コード長 1,168 bytes
コンパイル時間 4,212 ms
コンパイル使用メモリ 270,712 KB
実行使用メモリ 13,764 KB
最終ジャッジ日時 2024-09-23 03:58:18
合計ジャッジ時間 7,732 ms
ジャッジサーバーID
(参考情報)
judge5 / judge1
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 4
other AC * 3 TLE * 1 -- * 35
権限があれば一括ダウンロードができます

ソースコード

diff #

#pragma GCC target("avx2")
#pragma GCC optimize("O3")
#pragma GCC optimize("unroll-loops")
#include <bits/stdc++.h>
using namespace std;
int N,K;
pair<vector<int>,int> f(int n, int k, vector<int> &S, int x){
    vector<int> A={};
    int now = 0;
    for(int i=0;i<S.size();i++){
        if(i != 0) now++;
        if(S[i] == 1) continue;
        if(S[i] != 1 && (x+i)%2 == 1){
            if(now != 0) A.emplace_back(now);
            A.emplace_back(S[i]-1);
            now = 0;
        }
        else now += S[i]-1;
    }
    if(now != 0) A.emplace_back(now);
    return {A,1-(S[0] > 1 && x == 1)};
}
int main(){
    cin>>N>>K;
    vector<int> S={};
    for(int i=0;i<N;i++){
        char a;
        cin>>a;
        S.emplace_back(a-'0');
    }
    vector<int> T={};
    int now = 1;
    for(int i=1;i<N;i++){
      if(S[i] != S[i-1]){
            T.emplace_back(now);
            now=1;
        }
        else now++;
    }
    T.push_back(now);
    int a=S[0];
    for(int i=N;i>K;i--){
        pair<vector<int>,int> z = f(i,K,T,a);
        tie(T,a) = z;
    }
    for(int i=0;i<T.size();i++){
        for(int j=0;j<T[i];j++) cout<<(a+i)%2;
    }
    cout<<endl;
}
0