結果
| 問題 |
No.1150 シュークリームゲーム(Easy)
|
| コンテスト | |
| ユーザー |
qwewe
|
| 提出日時 | 2025-04-24 12:20:42 |
| 言語 | PyPy3 (7.3.15) |
| 結果 |
WA
|
| 実行時間 | - |
| コード長 | 2,617 bytes |
| コンパイル時間 | 309 ms |
| コンパイル使用メモリ | 82,380 KB |
| 実行使用メモリ | 106,360 KB |
| 最終ジャッジ日時 | 2025-04-24 12:20:58 |
| 合計ジャッジ時間 | 5,975 ms |
|
ジャッジサーバーID (参考情報) |
judge2 / judge3 |
(要ログイン)
| ファイルパターン | 結果 |
|---|---|
| sample | AC * 2 |
| other | AC * 2 WA * 41 |
ソースコード
import sys
from collections import deque
def main():
input = sys.stdin.read().split()
ptr = 0
N = int(input[ptr])
ptr += 1
s = int(input[ptr])
t = int(input[ptr + 1])
ptr += 2
A = list(map(int, input[ptr:ptr + N]))
ptr += N
adjacent = False
if abs(s - t) == 1 or (s == 1 and t == N) or (s == N and t == 1):
adjacent = True
if adjacent:
start = (s % N) + 1
end = t - 1 if t > 1 else N
remaining_vertices = []
if start <= end:
remaining_vertices = list(range(start, end + 1))
else:
part1 = list(range(start, N + 1))
part2 = list(range(1, end + 1))
remaining_vertices = part1 + part2
dq = deque([A[v - 1] for v in remaining_vertices])
x = 0
y = 0
turn = 0
while dq:
if turn == 0:
val = dq.popleft()
x += val
else:
val = dq.pop()
y += val
turn = 1 - turn
X = x + A[s - 1]
Y = y + A[t - 1]
result = X - Y
else:
def get_clock_path(s, t, N):
path = []
current = s
while True:
current = current % N + 1
if current == t:
break
path.append(current)
return path
def get_counter_clock_path(s, t, N):
path = []
current = s
while True:
current = (current - 2) % N + 1
if current == t:
break
path.append(current)
return path
path_clock = get_clock_path(s, t, N)
path_counter_clock = get_counter_clock_path(s, t, N)
dq_a = deque([A[v - 1] for v in path_clock])
dq_b = deque([A[v - 1] for v in path_counter_clock])
x = 0
y = 0
turn = 0
while dq_a or dq_b:
a_val = dq_a[0] if dq_a else -1
b_val = dq_b[0] if dq_b else -1
if a_val >= b_val and a_val != -1:
val = dq_a.popleft()
elif b_val != -1:
val = dq_b.popleft()
else:
if dq_a:
val = dq_a.popleft()
else:
val = dq_b.popleft()
if turn == 0:
x += val
else:
y += val
turn = 1 - turn
X = x + A[s - 1]
Y = y + A[t - 1]
result = X - Y
print(result)
if __name__ == '__main__':
main()
qwewe