結果

問題 No.2027 (1, 2, 3, …, N) 's Subset Sum
ユーザー ShirotsumeShirotsume
提出日時 2022-08-01 01:18:39
言語 PyPy3
(7.3.15)
結果
TLE  
実行時間 -
コード長 1,323 bytes
コンパイル時間 201 ms
コンパイル使用メモリ 82,340 KB
実行使用メモリ 307,772 KB
最終ジャッジ日時 2024-07-21 07:42:12
合計ジャッジ時間 4,521 ms
ジャッジサーバーID
(参考情報)
judge4 / judge5
このコードへのチャレンジ
(要ログイン)

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 41 ms
61,720 KB
testcase_01 AC 41 ms
53,744 KB
testcase_02 TLE -
testcase_03 -- -
testcase_04 -- -
testcase_05 -- -
testcase_06 -- -
testcase_07 -- -
testcase_08 -- -
testcase_09 -- -
testcase_10 -- -
testcase_11 -- -
testcase_12 -- -
testcase_13 -- -
testcase_14 -- -
testcase_15 -- -
testcase_16 -- -
testcase_17 -- -
testcase_18 -- -
testcase_19 -- -
testcase_20 -- -
testcase_21 -- -
testcase_22 -- -
testcase_23 -- -
testcase_24 -- -
testcase_25 -- -
testcase_26 -- -
権限があれば一括ダウンロードができます

ソースコード

diff #

import sys
input = lambda: sys.stdin.readline().rstrip()
ii = lambda: int(input())
mi = lambda: map(int, input().split())
li = lambda: list(mi())
inf = 2 ** 63 - 1
mod = 998244353
import random
n, s = mi()

class ListDict(object):
    def __init__(self):
        self.item_to_position = {}
        self.items = []

    def add_item(self, item):
        if item in self.item_to_position:
            return
        self.items.append(item)
        self.item_to_position[item] = len(self.items)-1

    def remove_item(self, item):
        position = self.item_to_position.pop(item)
        last_item = self.items.pop()
        if position != len(self.items):
            self.items[position] = last_item
            self.item_to_position[last_item] = position

    def choose_random_item(self):
        return random.choice(self.items)

summ = 0
ans = []
D = ListDict()

for i in range(1, n + 1):
    D.add_item(i)
cnt = 0
while True:
    cnt += 1
    now = D.choose_random_item()
    if summ + now <= s:
        summ += now
        ans.append(now)
        D.remove_item(now)
        if summ == s:
            break
    if cnt >= 10 ** 5 and summ != s:
        cnt = 0
        ans = []
        summ = 0
        D = ListDict()

        for i in range(1, n + 1):
            D.add_item(i)


print(len(ans))
print(*sorted(ans))

0