ABC082 C - Good Sequence

備忘録

問題

atcoder.jp

回答

import sys
import os
import math
import bisect
import collections
import itertools

ii = lambda: int(sys.stdin.buffer.readline().rstrip())
il = lambda: list(map(int, sys.stdin.buffer.readline().split()))
fl = lambda: list(map(float, sys.stdin.buffer.readline().split()))
iln = lambda n: [int(sys.stdin.buffer.readline().rstrip()) for _ in range(n)]

iss = lambda: sys.stdin.buffer.readline().decode().rstrip()
sl = lambda: list(map(str, sys.stdin.buffer.readline().decode().split()))
isn = lambda n: [sys.stdin.buffer.readline().decode().rstrip() for _ in range(n)]

lcm = lambda x, y: x * y / math.gcd(x, y)

MOD = 10 ** 9 + 7
MAX = float('inf')


def main():
    if os.getenv("LOCAL"):
        sys.stdin = open("input.txt", "r")

    X = ii()
    A = collections.Counter(il())

    ret = 0
    for k, v in A.items():
        if k > v:
            ret += v
        if k < v:
            ret += abs(k-v)
    print(ret)


if __name__ == '__main__':
    main()

考え方

整数列Aの各要素の個数を数え、
良い数列の条件を満たすように各要素を取り除く。
気を付ける点は要素を取り除くことしかできない点。
そのため、整数aの個数がxのとき、
a > xの場合には、x個全て取り除き、
a < xの場合には、x - a個取り除くことで回答を得られる。

実装はcollections.Counterを使用することで、
各要素の個数を高速に求めることが出来る。
あとはforで条件を満たすために必要な取り除く要素の個数を求めるだけ。