結果

問題 No.1369 交換門松列・竹
ユーザー Kude
提出日時 2021-01-29 23:25:03
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 323 ms / 2,000 ms
コード長 2,649 bytes
コンパイル時間 170 ms
コンパイル使用メモリ 82,304 KB
実行使用メモリ 85,376 KB
最終ジャッジ日時 2024-06-27 09:51:27
合計ジャッジ時間 4,475 ms
ジャッジサーバーID
(参考情報)
judge2 / judge1
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 1
other AC * 33
権限があれば一括ダウンロードができます

ソースコード

diff #

def solve():
    n = int(input())
    a = list(map(int, input().split()))
    for z in range(2):
        n = len(a)
        broken_adj = []
        broken_even = []
        broken_odd = []
        for i in range(n - 1):
            d = a[i+1] - a[i]
            if i & 1 == 0 and d <= 0 or i & 1 == 1 and d >= 0:
                broken_adj.append(i)
        for i in range(n - 2):
            if a[i] == a[i+2]:
                if i & 1 == 0:
                    broken_even.append(i)
                else:
                    broken_odd.append(i)
        if max(map(len, [broken_adj, broken_even, broken_odd])) <= 4:
            if broken_adj:
                s = broken_adj[0]
                t = s + 1
            elif broken_even:
                s = broken_even[0]
                t = s + 2
            else:
                s = broken_odd[0]
                t = s + 2
            for i in (s, t):
                for j in range(z, n):
                    if i == j:
                        continue
                    a[i], a[j] = a[j], a[i]
                    for k in (i, j):
                        if (0 <= k - 2 and a[k-2] == a[k] or
                            k + 2 < n and a[k] == a[k+2] or
                            k & 1 == 0 and (
                                0 <= k - 1 and a[k-1] <= a[k] or
                                k + 1 < n and a[k] >= a[k+1]) or
                            k & 1 == 1 and (
                                0 <= k - 1 and a[k-1] >= a[k] or
                                k + 1 < n and a[k] <= a[k+1]
                            )):
                            break
                    else:
                        ba = set(broken_adj)
                        be = set(broken_even)
                        bo = set(broken_odd)
                        for k in (i, j):
                            if k - 2 in bo:
                                bo.remove(k - 2)
                            if k - 2 in be:
                                be.remove(k - 2)
                            if k - 1 in ba:
                                ba.remove(k - 1)
                            if k in bo:
                                bo.remove(k)
                            if k in be:
                                be.remove(k)
                            if k in ba:
                                ba.remove(k)
                        if max(map(len, [ba, be, bo])) == 0:
                            print('Yes')
                            return
                    a[i], a[j] = a[j], a[i]
        if z == 0:
            a = [0] + a
    print('No')
    return

for _ in range(int(input())):
    solve()
0