提出 #65289220


ソースコード 拡げる

import sys
from tokenize import group

input = sys.stdin.readline

def count_numbers(numbers, a):
    count = 0
    for number in numbers:
        if number in a:
            count += 1
    return count

def main():
    N, D = [int(e) for e in input().split()]
    a = sorted([int(e) for e in input().split()])
    numbers_to_del = set()

    for i in range(len(a)-1):
        prev = a[i]
        if a[i] in numbers_to_del:
            continue
        for j in range(i+1, len(a)):
            cur = a[j]
            diff = cur - prev
            if diff == D:
                numbers_to_del.add(a[i])
                numbers_to_del.add(a[j])
            elif diff > D:
                break
    # print(a)
    numbers_to_del = list(numbers_to_del)

    groups = []
    while len(numbers_to_del) != 0:
        prev = numbers_to_del[0]-D
        group = []
        for e in numbers_to_del:
            if e - prev == D:
                group.append(e)
                prev = e
        groups.append(group)
        numbers_to_del = [x for x in numbers_to_del if x not in group]

    for group in groups:
        odd_numbers = group[::2]
        odd_count = count_numbers(odd_numbers, a)
        even_numbers = group[1::2]
        even_count = count_numbers(even_numbers, a)
        if odd_count > even_count:
            numbers_to_del += even_numbers
        else:
            numbers_to_del += odd_numbers

    # print(groups)
    # print(list(numbers_to_del))

    # count = 0
    print(count_numbers(numbers_to_del, a))
    # print(count)

main()

提出情報

提出日時
問題 D - Forbidden Difference
ユーザ Wasaw
言語 Python (PyPy 3.10-v7.3.12)
得点 0
コード長 1610 Byte
結果 WA
実行時間 2221 ms
メモリ 281056 KiB

ジャッジ結果

セット名 Sample All
得点 / 配点 0 / 0 0 / 425
結果
AC × 3
AC × 6
WA × 26
TLE × 8
セット名 テストケース
Sample 00_sample_01.txt, 00_sample_02.txt, 00_sample_03.txt
All 00_sample_01.txt, 00_sample_02.txt, 00_sample_03.txt, 01_random_01.txt, 01_random_02.txt, 01_random_03.txt, 01_random_04.txt, 01_random_05.txt, 01_random_06.txt, 01_random_07.txt, 01_random_08.txt, 01_random_09.txt, 01_random_10.txt, 01_random_11.txt, 01_random_12.txt, 01_random_13.txt, 01_random_14.txt, 01_random_15.txt, 01_random_16.txt, 01_random_17.txt, 01_random_18.txt, 01_random_19.txt, 01_random_20.txt, 01_random_21.txt, 01_random_22.txt, 01_random_23.txt, 01_random_24.txt, 01_random_25.txt, 01_random_26.txt, 01_random_27.txt, 01_random_28.txt, 01_random_29.txt, 01_random_30.txt, 01_random_31.txt, 01_random_32.txt, 02_handmade_01.txt, 02_handmade_02.txt, 02_handmade_03.txt, 02_handmade_04.txt, 02_handmade_05.txt
ケース名 結果 実行時間 メモリ
00_sample_01.txt AC 116 ms 83172 KiB
00_sample_02.txt AC 115 ms 82944 KiB
00_sample_03.txt AC 116 ms 83264 KiB
01_random_01.txt WA 382 ms 104444 KiB
01_random_02.txt WA 134 ms 85236 KiB
01_random_03.txt WA 1347 ms 104336 KiB
01_random_04.txt WA 471 ms 90384 KiB
01_random_05.txt TLE 2211 ms 103864 KiB
01_random_06.txt WA 321 ms 101552 KiB
01_random_07.txt WA 150 ms 103848 KiB
01_random_08.txt WA 218 ms 95048 KiB
01_random_09.txt WA 167 ms 103996 KiB
01_random_10.txt WA 166 ms 102080 KiB
01_random_11.txt WA 1080 ms 104220 KiB
01_random_12.txt WA 680 ms 99876 KiB
01_random_13.txt TLE 2212 ms 104396 KiB
01_random_14.txt WA 144 ms 92952 KiB
01_random_15.txt WA 163 ms 103848 KiB
01_random_16.txt WA 135 ms 89028 KiB
01_random_17.txt WA 1062 ms 105952 KiB
01_random_18.txt WA 868 ms 100836 KiB
01_random_19.txt WA 1077 ms 105624 KiB
01_random_20.txt WA 712 ms 100812 KiB
01_random_21.txt WA 1260 ms 105844 KiB
01_random_22.txt WA 1168 ms 105020 KiB
01_random_23.txt WA 1455 ms 108448 KiB
01_random_24.txt WA 665 ms 96116 KiB
01_random_25.txt TLE 2220 ms 263948 KiB
01_random_26.txt WA 208 ms 89144 KiB
01_random_27.txt TLE 2221 ms 281056 KiB
01_random_28.txt WA 244 ms 89632 KiB
01_random_29.txt TLE 2212 ms 105196 KiB
01_random_30.txt TLE 2212 ms 103276 KiB
01_random_31.txt TLE 2216 ms 181944 KiB
01_random_32.txt WA 389 ms 91336 KiB
02_handmade_01.txt TLE 2212 ms 104272 KiB
02_handmade_02.txt WA 140 ms 104344 KiB
02_handmade_03.txt AC 115 ms 83244 KiB
02_handmade_04.txt AC 114 ms 83160 KiB
02_handmade_05.txt AC 117 ms 83132 KiB