結果

問題 No.1694 ZerOne
ユーザー lam6er
提出日時 2025-04-15 22:12:49
言語 PyPy3
(7.3.15)
結果
TLE  
実行時間 -
コード長 1,534 bytes
コンパイル時間 259 ms
コンパイル使用メモリ 81,904 KB
実行使用メモリ 107,836 KB
最終ジャッジ日時 2025-04-15 22:14:14
合計ジャッジ時間 3,914 ms
ジャッジサーバーID
(参考情報)
judge2 / judge1
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
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)
    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