結果
| 問題 |
No.1171 Runs in Subsequences
|
| コンテスト | |
| ユーザー |
ayaoni
|
| 提出日時 | 2021-08-31 05:37:22 |
| 言語 | PyPy3 (7.3.15) |
| 結果 |
AC
|
| 実行時間 | 90 ms / 2,000 ms |
| コード長 | 1,291 bytes |
| コンパイル時間 | 405 ms |
| コンパイル使用メモリ | 82,444 KB |
| 実行使用メモリ | 102,704 KB |
| 最終ジャッジ日時 | 2024-11-24 20:31:18 |
| 合計ジャッジ時間 | 2,514 ms |
|
ジャッジサーバーID (参考情報) |
judge1 / judge4 |
(要ログイン)
| ファイルパターン | 結果 |
|---|---|
| sample | AC * 4 |
| other | AC * 18 |
ソースコード
import sys
def I(): return int(sys.stdin.readline().rstrip())
def MI(): return map(int,sys.stdin.readline().rstrip().split())
def LI(): return list(map(int,sys.stdin.readline().rstrip().split()))
def LI2(): return list(map(int,sys.stdin.readline().rstrip()))
def S(): return sys.stdin.readline().rstrip()
def LS(): return list(sys.stdin.readline().rstrip().split())
def LS2(): return list(sys.stdin.readline().rstrip())
S = S()
N = len(S)
mod = 10**9+7
power = [1] # power[i] = 2^i
for _ in range(N):
power.append((power[-1]*2) % mod)
T = [(-1,1)] # Sを縮約
prev = S[0]
count = 1
for s in S[1:]:
if s == prev:
count += 1
else:
T.append((ord(prev)-97,count))
prev = s
count = 1
T.append((ord(prev)-97,count))
M = len(T)-1
X = [0]*(M+1)
X[0] = 1
s = 0
for i in range(1,M+1):
a = T[i][1]
X[i] = (power[s]*(power[a]-1)) % mod
s += a
dp = [0]*(M+1)
# dp[i] = 左からi個目の文字で終わるような部分列に関する連の個数の和
S_dp = [0]*(M+1)
minus = [0]*26
s = 0
for i in range(1,M+1):
char,count = T[i]
x = (S_dp[i-1]+power[s]-minus[char])*(power[count]-1) % mod
dp[i] = x
S_dp[i] = (S_dp[i-1]+x) % mod
s += count
minus[char] += X[i]
minus[char] %= mod
ans = S_dp[-1]
print(ans)
ayaoni