結果

問題 No.2772 Appearing Even Times
コンテスト
ユーザー navel_tos
提出日時 2024-05-31 22:40:32
言語 PyPy3
(7.3.15)
結果
TLE  
実行時間 -
コード長 1,803 bytes
コンパイル時間 303 ms
コンパイル使用メモリ 82,404 KB
実行使用メモリ 195,564 KB
最終ジャッジ日時 2024-12-21 00:27:23
合計ジャッジ時間 62,600 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 7 TLE * 12
権限があれば一括ダウンロードができます

ソースコード

diff # 

#yukicoder 2772 Appearing Even Times

N = input()
MOD = 998244353

def brute(N: str):
    n = int(N)
    ans = 0
    for i in range(1, n + 1):
        i = str(i)
        C = [0] * 10
        for j in i:
            C[ int(j) ] ^= 1
        if not any(C):
            ans += 1
    return ans

def solve(N: str):
    #DP[i][f][g][S]: Nの下からi桁目まで見たとき、0から9までの数字の出現回数フラグがS、
    #                f = 以下フラグ
    #                g = 0, 1, 2: leading zerosがない、0が奇数個/偶数個つながる　状態数
    #next DP tech. で高速化
    DP = [[[0] * (1 << 10) for g in range(3)] for f in range(2)]
    N = N[::-1]

    #1桁目を埋める
    DP[1][1][1] = 1
    now = int(N[0])
    for k in range(1, 10):
        DP[k <= now][0][1 << k] += 1

    #2桁目以降を決定する
    for i, now in enumerate(N[1:], start = 1):
        now = int(now)
        nDP = [[[0] * (1 << 10) for g in range(3)] for f in range(2)]
        #0をつなげる場合
        for S in range(1 << 10):
            for f in range(2):
                nf = (now > 0) | f
                T = S ^ 1
                nDP[nf][1][T] += DP[f][0][S] + DP[f][2][S]
                nDP[nf][1][T] %= MOD
                nDP[nf][2][T] += DP[f][1][S]
                nDP[nf][2][T] %= MOD

        #1 - 9をつなげる場合
        for S in range(1 << 10):
            for k in range(1, 10):
                T = S ^ (1 << k)
                for f in range(2):
                    nf = (now > k) | ((now == k) & f)
                    nDP[nf][0][T] += sum(DP[f][g][S] for g in range(3))
                    nDP[nf][0][T] %= MOD
        DP, nDP = nDP, DP
    
    ans = DP[1][0][0] + DP[1][1][1] + DP[1][2][0] - 1
    ans %= MOD
    return ans

print( solve(N) )
0