結果

問題 No.2037 NAND Pyramid
コンテスト
ユーザー ああいい
提出日時 2022-08-14 12:46:08
言語 PyPy3
(7.3.15)
結果
WA  
実行時間 -
コード長 1,599 bytes
コンパイル時間 288 ms
コンパイル使用メモリ 82,560 KB
実行使用メモリ 256,116 KB
最終ジャッジ日時 2024-10-01 05:58:37
合計ジャッジ時間 5,231 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 6 WA * 33
権限があれば一括ダウンロードができます

ソースコード

diff # 

import sys


def calc(l):
    ans = []
    for i in range(len(l) - 1):
        ans.append(1 - (l[i] & l[i+1]))
    return ans

N,K = map(int,input().split())
K = N - K
S = input()
l = list(S)
l = list(map(int,l))
if K % 2 == 1:
    ll = []
    for i in range(N - 1):
        ll.append(1 - (l[i] & l[i+1]))
    l = ll
    K -= 1
    N -= 1
if K == 0:
    print(*l,sep = "")
    exit()

if K <= 20 or N <= 20:
    for _ in range(K):
        l = calc(l)
    print(*l,sep = "")
    exit()

u = K // 2
start = u
end = N - u - 1

ans = [0] * (N - K)

hidari = False
migi = False
for i in range(u):
    if l[i] == 0:
        hidari = True
        break
for i in range(end + 1,N):
    if l[i] == 0:
        migi = True
        break
"""
if l[start] == 1:
    if l[u+1] == 1:
        ans[0] = 1
    else:
        if hidari:
            ans[0] = 0
        else:
            ans[0] = 1
else:
    ans[0] = 0
if l[end] == 1:
    if l[end-1] == 1:
        ans[-1] = 1
    else:
        if migi:
            ans[-1] = 0
        else:
            ans[-1] = 1
else:
    ans[-1] = 0
"""
if l[start] == 1:
    if l[start + 1] == 1:
        ans[0] = 1
    else:
        if l[start - 1] == 1:
            ans[0] = 1
        else:
            ans[0] = 0
if l[end] == 1:
    if l[end - 1] == 0 and l[end] + 1 == 0:
        ans[-1] = 0
    else:
        ans[-1] = 1
for i in range(1,N - K - 1):
    if l[start+i] == 0:
        ans[i] = 0
        continue
    else:
        if l[start + i - 1] == 0 and l[start + i + 1] == 0:
            ans[i] = 0
        else:
            ans[i] = 1
print(*ans,sep = "")
print(start,end)
0