結果

問題 No.243 出席番号(2)
ユーザー titia
提出日時 2022-08-09 03:51:41
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
TLE  
実行時間 -
コード長 693 bytes
コンパイル時間 167 ms
コンパイル使用メモリ 12,672 KB
実行使用メモリ 16,384 KB
最終ジャッジ日時 2024-09-19 11:34:28
合計ジャッジ時間 13,061 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 24 TLE * 1 -- * 5
権限があれば一括ダウンロードができます

ソースコード

diff # 

# 包除原理の典型問題
# 解説を見てしまったが、これを解けないのはダメ。

mod=10**9+7

def main():
    N=int(input())
    A=[int(input()) for i in range(N)]
    C=[0]*N
    for a in A:
        if a<N:
            C[a]+=1

    DP=[0]*(N+1)
    DP[0]=1

    for v in C:
        if v==0:
            continue
        for i in range(N-1,-1,-1):
            DP[i+1]+=DP[i]*v
            DP[i+1]%=mod      
    ANS=0
    fac=1
    for i in range(N+1):
        if i%2==0:
            ANS+=fac*DP[N-i]
        else:
            ANS-=fac*DP[N-i]

        fac=fac*(i+1)%mod

        ANS%=mod

    if N%2==1:
        print((-ANS)%mod)
    else:
        print(ANS)
    
main()
0