Submission #46518796

Source Code Expand

import random
from collections import defaultdict
N, D, Q = map(int, input().split())
# 30≤N≤100
remainQ = Q
ans = [None]*N
ans = [i%D for i in range(N)] # random split
dict_idx = defaultdict(list)
for i, v in enumerate(ans):
    dict_idx[v].append(i)

while remainQ:
    num1, num2 = random.sample(range(D), 2) # どの集合を調整するか選ぶ
    if len(dict_idx[num1]) == 1 or len(dict_idx[num2]) == 1:continue
    # Interactive
    print(len(dict_idx[num1]), len(dict_idx[num2]), *dict_idx[num1], *dict_idx[num2], flush=True)
    ret = input()
    remainQ -= 1
    if ret == "=": continue
    if ret == "<":
        num1, num2 = num2, num1
    while remainQ:
        if len(dict_idx[num1]) == 1 or len(dict_idx[num2]) == 1: break
        # item = dict_idx[num1].pop() # 大きい方から取り出す
        item = random.choice(dict_idx[num1])
        dict_idx[num1].remove(item)
        dict_idx[num2].append(item)
        
        ans[item] = num2
        print(len(dict_idx[num1]), len(dict_idx[num2]), *dict_idx[num1], *dict_idx[num2], flush=True)
        ret = input()
        remainQ -= 1
        if ret != ">":
            break
print(*ans)

Submission Info

Submission Time
Task A - Balancing by Balance
User te1229
Language Python (PyPy 3.10-v7.3.12)
Score 2142046950
Code Size 1197 Byte
Status AC
Exec Time 185 ms
Memory 88008 KiB

Judge Result

Set Name test_ALL
Score / Max Score 2142046950 / 100000000000
Status
AC × 100
Set Name Test Cases
test_ALL test_0000.txt, test_0001.txt, test_0002.txt, test_0003.txt, test_0004.txt, test_0005.txt, test_0006.txt, test_0007.txt, test_0008.txt, test_0009.txt, test_0010.txt, test_0011.txt, test_0012.txt, test_0013.txt, test_0014.txt, test_0015.txt, test_0016.txt, test_0017.txt, test_0018.txt, test_0019.txt, test_0020.txt, test_0021.txt, test_0022.txt, test_0023.txt, test_0024.txt, test_0025.txt, test_0026.txt, test_0027.txt, test_0028.txt, test_0029.txt, test_0030.txt, test_0031.txt, test_0032.txt, test_0033.txt, test_0034.txt, test_0035.txt, test_0036.txt, test_0037.txt, test_0038.txt, test_0039.txt, test_0040.txt, test_0041.txt, test_0042.txt, test_0043.txt, test_0044.txt, test_0045.txt, test_0046.txt, test_0047.txt, test_0048.txt, test_0049.txt, test_0050.txt, test_0051.txt, test_0052.txt, test_0053.txt, test_0054.txt, test_0055.txt, test_0056.txt, test_0057.txt, test_0058.txt, test_0059.txt, test_0060.txt, test_0061.txt, test_0062.txt, test_0063.txt, test_0064.txt, test_0065.txt, test_0066.txt, test_0067.txt, test_0068.txt, test_0069.txt, test_0070.txt, test_0071.txt, test_0072.txt, test_0073.txt, test_0074.txt, test_0075.txt, test_0076.txt, test_0077.txt, test_0078.txt, test_0079.txt, test_0080.txt, test_0081.txt, test_0082.txt, test_0083.txt, test_0084.txt, test_0085.txt, test_0086.txt, test_0087.txt, test_0088.txt, test_0089.txt, test_0090.txt, test_0091.txt, test_0092.txt, test_0093.txt, test_0094.txt, test_0095.txt, test_0096.txt, test_0097.txt, test_0098.txt, test_0099.txt
Case Name Status Exec Time Memory
test_0000.txt AC 94 ms 84488 KiB
test_0001.txt AC 130 ms 85036 KiB
test_0002.txt AC 128 ms 86044 KiB
test_0003.txt AC 91 ms 84340 KiB
test_0004.txt AC 93 ms 84248 KiB
test_0005.txt AC 89 ms 84472 KiB
test_0006.txt AC 147 ms 85228 KiB
test_0007.txt AC 130 ms 85116 KiB
test_0008.txt AC 106 ms 84200 KiB
test_0009.txt AC 97 ms 84148 KiB
test_0010.txt AC 99 ms 84324 KiB
test_0011.txt AC 110 ms 84228 KiB
test_0012.txt AC 91 ms 84168 KiB
test_0013.txt AC 96 ms 84444 KiB
test_0014.txt AC 111 ms 84660 KiB
test_0015.txt AC 101 ms 84728 KiB
test_0016.txt AC 92 ms 84224 KiB
test_0017.txt AC 95 ms 84212 KiB
test_0018.txt AC 126 ms 85320 KiB
test_0019.txt AC 92 ms 84644 KiB
test_0020.txt AC 136 ms 85160 KiB
test_0021.txt AC 102 ms 84144 KiB
test_0022.txt AC 104 ms 84840 KiB
test_0023.txt AC 97 ms 84528 KiB
test_0024.txt AC 119 ms 84764 KiB
test_0025.txt AC 101 ms 84240 KiB
test_0026.txt AC 129 ms 85360 KiB
test_0027.txt AC 91 ms 84352 KiB
test_0028.txt AC 96 ms 84468 KiB
test_0029.txt AC 127 ms 85680 KiB
test_0030.txt AC 93 ms 84164 KiB
test_0031.txt AC 89 ms 84476 KiB
test_0032.txt AC 106 ms 84188 KiB
test_0033.txt AC 99 ms 84276 KiB
test_0034.txt AC 132 ms 84648 KiB
test_0035.txt AC 92 ms 84620 KiB
test_0036.txt AC 92 ms 84164 KiB
test_0037.txt AC 117 ms 84584 KiB
test_0038.txt AC 106 ms 84544 KiB
test_0039.txt AC 114 ms 84360 KiB
test_0040.txt AC 142 ms 85492 KiB
test_0041.txt AC 111 ms 84752 KiB
test_0042.txt AC 99 ms 84604 KiB
test_0043.txt AC 140 ms 85320 KiB
test_0044.txt AC 119 ms 84712 KiB
test_0045.txt AC 98 ms 84308 KiB
test_0046.txt AC 102 ms 84572 KiB
test_0047.txt AC 133 ms 84632 KiB
test_0048.txt AC 129 ms 84924 KiB
test_0049.txt AC 101 ms 84468 KiB
test_0050.txt AC 105 ms 84376 KiB
test_0051.txt AC 104 ms 84076 KiB
test_0052.txt AC 165 ms 86052 KiB
test_0053.txt AC 102 ms 84508 KiB
test_0054.txt AC 93 ms 84420 KiB
test_0055.txt AC 91 ms 84644 KiB
test_0056.txt AC 105 ms 84228 KiB
test_0057.txt AC 106 ms 84484 KiB
test_0058.txt AC 102 ms 84292 KiB
test_0059.txt AC 119 ms 84748 KiB
test_0060.txt AC 126 ms 85000 KiB
test_0061.txt AC 92 ms 84692 KiB
test_0062.txt AC 134 ms 85696 KiB
test_0063.txt AC 113 ms 84576 KiB
test_0064.txt AC 91 ms 84372 KiB
test_0065.txt AC 92 ms 84376 KiB
test_0066.txt AC 183 ms 88008 KiB
test_0067.txt AC 96 ms 84712 KiB
test_0068.txt AC 113 ms 84568 KiB
test_0069.txt AC 98 ms 84860 KiB
test_0070.txt AC 102 ms 84472 KiB
test_0071.txt AC 111 ms 84744 KiB
test_0072.txt AC 130 ms 85472 KiB
test_0073.txt AC 101 ms 84124 KiB
test_0074.txt AC 117 ms 84424 KiB
test_0075.txt AC 93 ms 84444 KiB
test_0076.txt AC 101 ms 84728 KiB
test_0077.txt AC 103 ms 84692 KiB
test_0078.txt AC 103 ms 84572 KiB
test_0079.txt AC 99 ms 84472 KiB
test_0080.txt AC 91 ms 84304 KiB
test_0081.txt AC 97 ms 84540 KiB
test_0082.txt AC 95 ms 84716 KiB
test_0083.txt AC 95 ms 84652 KiB
test_0084.txt AC 116 ms 84348 KiB
test_0085.txt AC 96 ms 84616 KiB
test_0086.txt AC 92 ms 84492 KiB
test_0087.txt AC 185 ms 87584 KiB
test_0088.txt AC 165 ms 86952 KiB
test_0089.txt AC 136 ms 85256 KiB
test_0090.txt AC 104 ms 84532 KiB
test_0091.txt AC 114 ms 84744 KiB
test_0092.txt AC 120 ms 84292 KiB
test_0093.txt AC 94 ms 84340 KiB
test_0094.txt AC 110 ms 84568 KiB
test_0095.txt AC 93 ms 84140 KiB
test_0096.txt AC 127 ms 85088 KiB
test_0097.txt AC 116 ms 84764 KiB
test_0098.txt AC 105 ms 84484 KiB
test_0099.txt AC 102 ms 84144 KiB