QOJ.ac

QOJ

IDProblemSubmitterResultTimeMemoryLanguageFile sizeSubmit timeJudge time
#676710#8790. First BillionmalkovskyTL 1196ms54880kbPython32.0kb2024-10-25 23:24:092024-10-25 23:24:11

Judging History

你现在查看的是最新测评结果

  • [2024-10-25 23:24:11]
  • 评测
  • 测评结果:TL
  • 用时:1196ms
  • 内存:54880kb
  • [2024-10-25 23:24:09]
  • 提交

answer


def knapsap_mod(nums):
    mods = [1001, 2003, 1003, 1997]
    targets = [1000000000 % mod for mod in mods]
    dp = [[[0, 0, 0, 0] for i in range(2003)] for _ in range(len(nums) + 1)]
    dp[0][0][0] = 1
    dp[0][0][1] = 1
    dp[0][0][2] = 1
    dp[0][0][3] = 1


    for i in range(len(nums)):
        for k in range(4):
            for j in range(mods[k]):
                if dp[i][j][k] == 1:
                    dp[i + 1][(j + nums[i]) % mods[k]][k] = 1
                    dp[i + 1][j][k] = 1

    n = len(nums)

    valid = {tuple(targets): []}
    while n > 0:
        new_valid = dict()
        # print(nums[n - 1])
        # print(len(valid))
        for valid_tuple, result in valid.items():
            left = tuple([(target + (mod - (nums[n - 1] % mod))) % mod for target, mod in zip(valid_tuple, mods)])

            ok = max([dp[n - 1][valid_tuple[k]][k] for k in range(4)])

            if ok == 1:
                new_valid[valid_tuple] = [x for x in result]
            ok = max([dp[n - 1][left[k]][k] for k in range(4)])
            if ok == 1:
                result.append(n - 1)
                new_valid[left] = [x for x in result]

        n -= 1
        valid = new_valid

    # print(sorted(valid))
    # if (0, 0, 0, 0) in valid:
    #     print(valid[(0, 0, 0, 0)])

    return valid[(0, 0, 0, 0)]


def __main__():
    n = int(input())
    numbers = list(map(int, input().split()))
    target = 1000000000
    # print(target % 1001, target % 1003, target % 2003)

    answer = []
    result = knapsap_mod(numbers)
    # print(result, sum([numbers[i] for i in result]))
    # print(sum([numbers[i] for i in result]) % 1001, sum([numbers[i] for i in result]) % 1003, sum([numbers[i] for i in result]) % 2003)
    if sum([numbers[i] for i in result]) == target:
        answer = result


        # print(num_at_least_k)

    print(len(answer), end=" ")
    for num in answer:
        print(num + 1, end=" ")


__main__()

Details

Tip: Click on the bar to expand more detailed information

Test #1:

score: 100
Accepted
time: 12ms
memory: 12784kb

input:

10
386413329 88494216 245947398 316438989 192751270 204627269 65749456 3938400 150458676 345180997

output:

5 10 8 4 3 2 

result:

ok OK (n = 10)

Test #2:

score: 0
Accepted
time: 18ms
memory: 12784kb

input:

10
119486233 299942886 169540407 349937991 597883752 32230162 140514533 57341098 12602102 220520836

output:

5 10 7 4 3 1 

result:

ok OK (n = 10)

Test #3:

score: 0
Accepted
time: 13ms
memory: 13616kb

input:

14
384615281 84612238 83310504 54746763 142296081 56775470 128760350 343006424 177232390 214368720 67220468 21895072 16352717 224807522

output:

7 14 11 8 5 4 3 2 

result:

ok OK (n = 14)

Test #4:

score: 0
Accepted
time: 27ms
memory: 13716kb

input:

14
270208635 14270307 89661499 113578022 47687195 101043954 38775146 208193324 650676076 351701957 3427619 59535626 24230888 27009752

output:

7 14 13 11 9 8 7 5 

result:

ok OK (n = 14)

Test #5:

score: 0
Accepted
time: 81ms
memory: 17268kb

input:

20
61638928 106712373 5946815 178135484 4937573 111395400 15504655 67139983 101814514 312223647 130341028 43244171 37671364 54108486 337181317 37924824 153793862 70383750 102917244 66984582

output:

10 20 19 17 16 15 7 6 3 2 1 

result:

ok OK (n = 20)

Test #6:

score: 0
Accepted
time: 79ms
memory: 17116kb

input:

20
67858098 61231428 99398662 1883806 82465954 303619377 87516412 154956240 94872199 76508350 13276828 136541811 203282099 99160366 127539385 13364660 141176136 39751629 67888657 127707903

output:

10 20 19 18 16 15 10 9 7 6 2 

result:

ok OK (n = 20)

Test #7:

score: 0
Accepted
time: 1158ms
memory: 54772kb

input:

24
17125795 281143405 10375259 196293002 158174864 34520650 52919232 87393970 99085271 62281508 67168428 55174991 54533464 51393059 89276370 41441658 72793517 30466999 73758332 97064918 111541434 142047546 12934221 101092107

output:

12 24 21 16 14 12 11 9 8 7 6 2 1 

result:

ok OK (n = 24)

Test #8:

score: 0
Accepted
time: 1196ms
memory: 54880kb

input:

24
70224368 148769600 36654748 23404220 15009825 57449487 46896672 6065662 10377031 133719710 23220853 184445684 8462667 88501546 155244839 229323557 140109402 52520271 78995771 75721556 87987586 118427778 107013825 101453342

output:

12 24 22 18 16 15 14 13 9 8 6 4 2 

result:

ok OK (n = 24)

Test #9:

score: -100
Time Limit Exceeded

input:

28
122321206 60841271 22767116 183943582 6247754 32767541 19129802 21313874 144503909 59360441 12259051 19044256 50267333 25766572 133411289 32253746 102412217 46186594 55413161 39907615 52325783 86862071 185310732 138228874 22000146 149813853 98156445 77183766

output:


result: