結果

問題 No.3370 AB → BA
コンテスト
ユーザー Aralov Otabek
提出日時 2026-01-08 23:11:31
言語 PyPy3
(7.3.17)
結果
TLE  
実行時間 -
コード長 1,016 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 280 ms
コンパイル使用メモリ 82,688 KB
実行使用メモリ 267,692 KB
最終ジャッジ日時 2026-01-08 23:11:35
合計ジャッジ時間 4,161 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 1
other TLE * 1 -- * 19
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

import sys

MOD = 998244353

def main():
    data = sys.stdin.read().strip()
    if not data:
        return
    
    # A harflarini hisoblash
    B = []
    cnt = 0
    for i, ch in enumerate(data, 1):
        if ch == 'A':
            B.append(i - cnt)
            cnt += 1
    
    if not B:
        print(1)
        return
    
    m = len(B)
    
    # DP jadvali
    max_val = B[-1]
    dp = [0] * (max_val)
    
    # Boshlang'ich to'ldirish
    for j in range(B[0]):
        dp[j] = 1
    
    # Har bir keyingi A uchun
    for i in range(1, m):
        end = B[i]
        
        # Prefix sumni hisoblash va yangi dp ni bir vaqtda
        new_dp = [0] * max_val
        current_sum = 0
        
        for j in range(end):
            current_sum = (current_sum + dp[j]) % MOD
            new_dp[j] = current_sum
        
        dp = new_dp
    
    # Natija
    result = 0
    for j in range(max_val):
        result = (result + dp[j]) % MOD
    
    print(result)

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