QOJ.ac
QOJ
ID | Problem | Submitter | Result | Time | Memory | Language | File size | Submit time | Judge time |
---|---|---|---|---|---|---|---|---|---|
#124244 | #5665. AA Country and King Dreamoon | bakuganulit | RE | 5ms | 8032kb | Python3 | 1.4kb | 2023-07-14 14:51:58 | 2023-07-14 14:51:59 |
Judging History
answer
def solve():
n = int(input())
path = list(map(int, input().split()))
path[-1] = 1
path[0] = 1
available = set(range(1, n + 1))
for num in path:
if num in available:
available.remove(num)
zeros = path.count(0)
last_city_index = path.index(0) - 1
last_city = path[last_city_index]
next_cities = [x for x in path[last_city_index+1:] if x != 0]
if next_cities:
next_city = next_cities[0]
else:
next_city = 0
missing_path = []
if zeros % 2 == 0:
if last_city > next_city:
for i in range(zeros):
if i % 2 == 0:
missing_path.append(next_city)
else:
missing_path.append(available.pop())
elif last_city < next_city:
for i in range(zeros):
if i % 2 == 0:
missing_path.append(available.pop())
else:
missing_path.append(last_city)
else:
for i in range(zeros):
if i % 2 == 0:
missing_path.append(available.pop())
else:
missing_path.append(last_city)
missing_path = missing_path[::-1]
ans = [x if x != 0 else missing_path.pop() for x in path]
print(" ".join(str(num) for num in ans))
if __name__ == "__main__":
t = int(input())
for _ in range(t):
solve()
Details
Tip: Click on the bar to expand more detailed information
Test #1:
score: 100
Accepted
time: 5ms
memory: 8032kb
input:
9 5 1 2 3 2 0 2 1 5 1 5 1 2 3 0 0 2 1 5 1 5 1 2 0 0 0 2 1 5 1 5 1 2 0 0 0 0 1 5 1 5 1 0 0 0 0 0 1 5 1 5 1 0 0 0 0 0 0 5 1 5 1 0 0 0 0 0 0 0 1 5 1 0 0 0 0 0 0 0 0 5 0 0 0 0 0 0 0 0 0
output:
1 2 3 2 4 2 1 5 1 1 2 3 2 4 2 1 5 1 1 2 3 2 4 2 1 5 1 1 2 1 3 1 4 1 5 1 1 2 1 3 1 4 1 5 1 1 2 1 3 1 4 1 5 1 1 2 1 3 1 4 1 5 1 1 2 1 3 1 4 1 5 1 1 2 1 3 1 4 1 5 1
result:
ok 9 lines
Test #2:
score: -100
Runtime Error
input:
28668 2 0 2 1 2 0 0 1 2 0 0 0 2 1 0 1 2 1 0 0 2 1 2 0 3 0 2 1 3 1 3 0 0 1 3 1 3 0 0 0 3 1 3 0 0 0 0 1 3 0 0 0 0 0 3 1 0 1 3 1 3 1 0 0 3 1 3 1 0 0 0 1 3 1 0 0 0 0 3 1 2 0 3 1 3 1 2 0 0 1 3 1 2 0 0 0 3 1 2 1 0 1 3 1 2 1 0 0 3 1 2 1 3 0 3 0 2 3 2 1 3 0 0 3 2 1 3 0 0 0 2 1 3 1 0 3 2 1 3 1 0 0 2 1 3 1 2 ...