結果

問題 No.1694 ZerOne
コンテスト
ユーザー gew1fw
提出日時 2025-06-12 13:37:12
言語 PyPy3
(7.3.15)
結果
TLE  
実行時間 -
コード長 1,758 bytes
コンパイル時間 388 ms
コンパイル使用メモリ 82,592 KB
実行使用メモリ 55,444 KB
最終ジャッジ日時 2025-06-12 13:42:57
合計ジャッジ時間 4,419 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other TLE * 1 -- * 30
権限があれば一括ダウンロードができます

ソースコード

diff # 

from collections import deque

def count_possible_strings(S):
    visited = set()
    queue = deque([S])
    visited.add(S)
    
    while queue:
        s = queue.popleft()
        n = len(s)
        # Precompute count of 0s and 1s for all possible substrings
        cnt = [[(0, 0) for _ in range(n)] for __ in range(n)]
        for i in range(n):
            zeros = 0
            ones = 0
            for j in range(i, n):
                if s[j] == '0':
                    zeros += 1
                else:
                    ones += 1
                cnt[i][j] = (zeros, ones)
        
        # Generate all possible t and u pairs
        for t_start in range(n):
            for t_end in range(t_start, n):
                t_zeros, t_ones = cnt[t_start][t_end]
                # u must start after t_end
                for u_start in range(t_end + 1, n):
                    for u_end in range(u_start, n):
                        u_zeros, u_ones = cnt[u_start][u_end]
                        if t_zeros == u_zeros and t_ones == u_ones:
                            # Swap t and u
                            # parts: [0..t_start-1] + u + [t_end+1..u_start-1] + t + [u_end+1..n-1]
                            part1 = s[:t_start]
                            part2 = s[u_start:u_end+1]
                            part3 = s[t_end+1:u_start]
                            part4 = s[t_start:t_end+1]
                            part5 = s[u_end+1:]
                            new_s = part1 + part2 + part3 + part4 + part5
                            if new_s not in visited:
                                visited.add(new_s)
                                queue.append(new_s)
    return len(visited)

S = input().strip()
print(count_possible_strings(S))
0