ABC082 C - Good Sequence
備忘録
問題
回答
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
で条件を満たすために必要な取り除く要素の個数を求めるだけ。