結果

問題 No.1171 Runs in Subsequences
ユーザー WSKRWSKR
提出日時 2020-09-22 21:15:35
言語 Python3
(3.12.2 + numpy 1.26.4 + scipy 1.12.0)
結果
TLE  
実行時間 -
コード長 1,175 bytes
コンパイル時間 101 ms
コンパイル使用メモリ 11,028 KB
実行使用メモリ 489,176 KB
最終ジャッジ日時 2023-09-09 03:38:15
合計ジャッジ時間 4,315 ms
ジャッジサーバーID
（参考情報） 		judge12 / judge11
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 16 ms
12,388 KB
testcase_01 AC 16 ms
7,980 KB
testcase_02 AC 17 ms
8,120 KB
testcase_03 AC 18 ms
8,240 KB
testcase_04 AC 16 ms
8,144 KB
testcase_05 AC 17 ms
8,004 KB
testcase_06 AC 17 ms
8,060 KB
testcase_07 AC 17 ms
8,084 KB
testcase_08 AC 17 ms
8,064 KB
testcase_09 AC 17 ms
8,028 KB
testcase_10 AC 36 ms
10,476 KB
testcase_11 AC 36 ms
10,452 KB
testcase_12 AC 37 ms
10,464 KB
testcase_13 AC 37 ms
10,596 KB
testcase_14 AC 37 ms
10,520 KB
testcase_15 TLE -
testcase_16 -- -
testcase_17 -- -
testcase_18 -- -
testcase_19 -- -
testcase_20 -- -
testcase_21 -- -
権限があれば一括ダウンロードができます

ソースコード

diff # 

#yukicoder practice

#runs in subsequences

mod = 10**9 + 7

alpha_list = list("abcdefghijklmnopqrstuvwxyz")

num_dict = {alpha_list[i]: i for i, v in enumerate(alpha_list)}


def to_num(string):
    return num_dict[string]


def main():
    s = list(input())
    N = len(s)

    dp = [[[0, 0] for j in range(26)] for i in range(N)]
    sumdp = [[0, 0] for i in range(N)]
    #dp[index][alpha] = [sum_of_len,group_count]
    a = list(map(to_num, s))

    dp[0][a[0]] = [1, 1]
    sumdp[0] = [1, 1]
    for i in range(1, N):
        num = a[i]
        sum_len, group_cnt = sumdp[i-1]
        for j in range(26):
            if j != num:
                dp[i][j][0] = dp[i-1][j][0]
                dp[i][j][1] = dp[i-1][j][1]

            else:
                dp[i][j][0] = (sum_len + dp[i-1][j][0] + 1 +
                               group_cnt - dp[i-1][j][1]) % mod
                dp[i][j][1] = (group_cnt + dp[i-1][j][1] + 1) % mod

            sumdp[i][0] += dp[i][j][0]
            sumdp[i][1] += dp[i][j][1]
            sumdp[i][0] %= mod
            sumdp[i][1] %= mod

    ans = sumdp[N-1][0]
    print(ans % mod)
    return


if __name__ == "__main__":
    main()
0