結果

問題 No.2102 [Cherry Alpha *] Conditional Reflection
ユーザー Akijin_007Akijin_007
提出日時 2022-10-14 22:53:08
言語 PyPy3
(7.3.15)
結果
WA  
実行時間 -
コード長 1,679 bytes
コンパイル時間 201 ms
コンパイル使用メモリ 82,400 KB
実行使用メモリ 282,836 KB
最終ジャッジ日時 2024-06-26 16:41:11
合計ジャッジ時間 45,717 ms
ジャッジサーバーID
(参考情報)
judge4 / judge5
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
other AC * 27 WA * 43
権限があれば一括ダウンロードができます

ソースコード

diff #

#int(input())
#map(int, input().split())
#list(map(int, input().split()))

N = int(input())

S = [0] * N
for i in range(N):
    S[i] = input()

d = dict()
ans = [0] * N

def check(x, y):
    if len(x) != len(y):
        return 0
    dif = []
    for i in range(len(x)):
        if x[i] != y[i]:
            dif.append(i)
    if len(dif) == 0:
        return 1
    elif len(dif) == 2:
        u = dif[0]
        v = dif[1]
        if u + 1 == v and x[u] == y[v] and x[v] == y[u]:
            return 1
        else:
            return 0
    else:
        return 0

for i in range(N):
    a = []
    u, v = divmod(len(S[i]), 2)
    u += v
    for j in range(u):
        s = S[i][2*j:2*j+2]
        if len(s) == 1:
            a.append(s)
            continue
        if s[0] < s[1]:
            a.append(s)
        else:
            a.append(s[1] + s[0])
    t = tuple(a)
    if t not in d:
        d[t] = [i]
    else:
        for x in d[t]:
            if check(S[i], S[x]):
                ans[i] = 1
                break
        d[t].append(i)

d = dict()

for i in range(N):
    if ans[i] == 1:
        continue
    a = [S[i][0]]
    u, v = divmod(len(S[i]), 2)
    for j in range(u):
        s = S[i][2*j+1:2*j+3]
        if len(s) == 1:
            a.append(s)
            continue
        if s[0] < s[1]:
            a.append(s)
        else:
            a.append(s[1] + s[0])
    t = tuple(a)
    if t not in d:
        d[t] = [i]
    else:
        for x in d[t]:
            if check(S[i], S[x]):
                ans[i] = 1
                break
        d[t].append(i)

# print(d)

for i in range(N):
    if ans[i] == 1:
        print("Yes")
    else:
        print("No")
0