結果

問題 No.1694 ZerOne
コンテスト
ユーザー lam6er
提出日時 2025-04-15 22:11:09
言語 PyPy3
(7.3.15)
結果
TLE  
実行時間 -
コード長 1,534 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 152 ms
コンパイル使用メモリ 81,712 KB
実行使用メモリ 110,636 KB
最終ジャッジ日時 2025-04-15 22:12:42
合計ジャッジ時間 3,719 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other TLE * 1 -- * 30
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

from collections import deque

def count_possible_strings(s):
    visited = set()
    queue = deque([s])
    visited.add(s)
    n = len(s)
    
    while queue:
        current = queue.popleft()
        # Iterate through all possible t substrings
        for t_start in range(n):
            for t_end in range(t_start, n):
                t = current[t_start:t_end+1]
                t0 = t.count('0')
                t1 = len(t) - t0
                # Iterate through all possible u substrings after t
                for u_start in range(t_end + 1, n):
                    for u_end in range(u_start, n):
                        u = current[u_start:u_end+1]
                        u0 = u.count('0')
                        u1 = len(u) - u0
                        # Check if t and u have the same 0s and 1s and same length
                        if t0 == u0 and t1 == u1 and (t_end - t_start) == (u_end - u_start):
                            # Create the new string by swapping t and u
                            new_str = (
                                current[:t_start] +
                                u +
                                current[t_end+1:u_start] +
                                t +
                                current[u_end+1:]
                            )
                            if new_str not in visited:
                                visited.add(new_str)
                                queue.append(new_str)
    return len(visited)

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