結果
| 問題 |
No.3047 Verification of Sorting Network
|
| ユーザー |
👑  Mizar
|
| 提出日時 | 2025-02-12 19:07:28 |
| 言語 | Python3 (3.13.1 + numpy 2.2.1 + scipy 1.14.1) |
| 結果 |
TLE
|
| 実行時間 | - |
| コード長 | 8,046 bytes |
| コンパイル時間 | 640 ms |
| コンパイル使用メモリ | 12,672 KB |
| 実行使用メモリ | 19,876 KB |
| 最終ジャッジ日時 | 2025-03-05 20:40:12 |
| 合計ジャッジ時間 | 3,851 ms |
|
ジャッジサーバーID （参考情報） |
judge2 / judge1 |
（要ログイン）
| ファイルパターン | 結果 |
|---|---|
| sample | AC * 2 TLE * 1 |
| other | -- * 61 |
ソースコード
"""
yukicoder Problem: Verify Sorting Network
"""
import functools
import math
import sys
SHOW_PROGRESS = False
UNLIMITED = False
MAX_TESTCASES = 1000
MAX_N = 27
MAX_COST = 1e8
class IsSortingOk:
"""is_sorting_network Ok type"""
def __init__(self, value: list[bool]):
self.value = value
def __bool__(self):
return True
def __str__(self):
return 'Yes'
def get_data(self):
"""get data value"""
return self.value
class IsSortingNg:
"""is_sorting_network Ng type"""
def __init__(self, value: list[bool]):
self.value = value
def __bool__(self):
return False
def __str__(self):
return 'No'
def get_error(self):
"""get error value"""
return self.value
def fib1(n: int) -> list[int]:
"""フィボナッチ数列 [1,1,2,3,…,Fib(n+1)] を生成します。"""
return functools.reduce(lambda x, _: x + [sum(x[-2:])], range(n), [1])
def is_sorting_network(n: int, net: list[tuple[int, int]], show_progress=False) -> IsSortingOk | IsSortingNg:
"""
与えられたネットワークが sorting network であるかどうかを調べます。
時間計算量 O(m * phi**n) で動作します。 phi は黄金比1.618...です。
ref: wikipedia: Sorting network
https://ja.wikipedia.org/wiki/%E3%82%BD%E3%83%BC%E3%83%86%E3%82%A3%E3%83%B3%E3%82%B0%E3%83%8D%E3%83%83%E3%83%88%E3%83%AF%E3%83%BC%E3%82%AF
ref: Hisayasu Kuroda. (1997). A proposal of Gap Decrease Sorting Network.
Trans.IPS.Japan, vol.38, no.3, p.381-389.
http://id.nii.ac.jp/1001/00013442/
ref: brianpursley/sorting-network, PR#9 three-valued-logic DFS approach
https://github.com/brianpursley/sorting-network/pull/9
"""
assert 2 <= n
# 0-indexed 入力の範囲を確認
assert all(0 <= a < b < n for a, b in net)
# 比較器の数
m = len(net)
# 初期状態はすべて '?' = 不定: 0 または 1 に決定されていない
stack = [((1 << n) - 1, (1 << n) - 1, 0)]
# 使われる事がある比較器かどうかを記録
unused = [True] * m
# ソートされない位置を記録
unsorted_i = 0
# フィボナッチ数列を生成
fib = fib1(n)
# 探索枝の進捗: 0 から fib[n] まで
progress, next_progress = 0, 0
# 'a: 分岐スタックのループ
while stack:
# 分岐スタックを取得
# z: 0状態ベクトル
# o: 1状態ベクトル
# i: 比較器のインデックス
z, o, i = stack.pop()
# 'b: 比較器のループ
while i < m:
# 比較器を取得
a, b = net[i]
i += 1
# すべての分岐で p が '0...01...1' または '0...0?1...1' にソートされるかどうかを確認します。
if ((o >> a) & 1) == 0 or ((z >> b) & 1) == 0:
pass # p[a]=='0' もしくは p[b]=='1' の場合、何もしない
elif ((z >> a) & 1) == 1 and ((o >> b) & 1) == 1:
# 入力(p[a],p[b]) == (?,?) の場合、
# (p[a],p[b]) が (1,0) となる可能性を内包しているため、交換が起きる入力が有り得る
unused[i - 1] = False
# 3値論理の組1つでは比較交換器の出力を正確に表現できないため分岐を作ります:
# 2値論理で表現すると出力は: {(0,0),(0,1),(1,1)}
# 3値論理で表現すると: {(0,0),(?,1)} または {(0,?),(1,1)}
# この実装では {(0,0),(?,1)} の分岐を作ります。
qz, qo = z, o
# qo = (qo & ~(1 << a) & ~(1 << b))
qo ^= (1 << a) ^ (1 << b)
z ^= (1 << b)
# 分岐 (?,?) → (0,0),(?,1)
# 'qz,qo' が '0...01...1' or '0...0?1...1' にソートされていれば進捗を加算
if (qo & (qz >> 1)) == 0:
progress += 1
else:
# 'qz,qo' が '0...01...1' or '0...0?1...1' にソートされていなければスタックに追加します。
stack.append((qz, qo, i))
# z,o が '0...01...1' にソートされていれば次の分岐へ
if (o & (z >> 1)) == 0:
progress += 1
break # 'a: 次の分岐へ
else:
# p[a]!='0' かつ p[b]!='1' かつ (p[a],p[b])!=('?','?') の場合
# (p[a],p[b]) が (1,0) となる可能性を内包しているため、交換が起きる入力が有り得る
unused[i - 1] = False
# (p[a],p[b]) が [(?,0),(1,0),(1,?)] の場合、
# (p[a],p[b]) → (p[b],p[a]) のように交換します。
xz = ((z >> a) ^ (z >> b)) & 1
xo = ((o >> a) ^ (o >> b)) & 1
z ^= ((xz << a) | (xz << b))
o ^= ((xo << a) | (xo << b))
# z,o が '0...01...1' or '0...0?1...1' にソートされていれば次の分岐へ
if (o & (z >> 1)) == 0:
progress += 1
break
else:
# すべての比較器を使用しても p がソートされていない分岐がある場合
# ソートされない位置を記録
unsorted_i |= (o & (z >> 1))
# 残り未知数に応じた進捗を加算
progress += fib[(z & o).bit_count()]
# 進捗を表示
if show_progress and progress >= next_progress:
percent = progress * 100 // fib[n]
sys.stderr.write(f'{percent}%\r')
# 1% 進むごとに更新: ceil((percent + 1) * fib[n] / 100)
next_progress = ((percent + 1) * fib[n] - 1) // 100 + 1
if show_progress:
sys.stderr.write('\n')
# 探索枝の数がフィボナッチ数列の値と一致することを確認
assert progress == fib[n]
# ソートされていない分岐がある場合
if unsorted_i != 0:
unsorted = [((unsorted_i >> i) & 1) != 0 for i in range(n - 1)]
return IsSortingNg(unsorted)
# すべての分岐でソートされている場合
return IsSortingOk(unused)
# 黄金比 (1+sqrt(5))/2 ≒ 1.618033988749895
PHI = math.sqrt(1.25) + 0.5 # 黄金比
def main():
"""テストケースの入出力処理"""
t = int(sys.stdin.readline())
assert t <= MAX_TESTCASES or UNLIMITED
cost = 0
for _ in range(t):
n, m = map(int, sys.stdin.readline().split())
assert 2 <= n <= MAX_N or UNLIMITED
assert 1 <= m <= n * (n - 1) // 2 or UNLIMITED
cost += m * PHI**n # テストケースの計算量コスト
assert cost <= MAX_COST or UNLIMITED
# 1-indexed -> 0-indexed
a = map(lambda x: int(x) - 1, sys.stdin.readline().split())
b = map(lambda x: int(x) - 1, sys.stdin.readline().split())
cmps: list[tuple[int, int]] = list(zip(a, b))
assert len(cmps) == m
assert all(0 <= a < b < n for a, b in cmps)
# is_sorting: 与えられたネットワークが sorting network であるかどうか
is_sorting = is_sorting_network(n, cmps, show_progress=SHOW_PROGRESS)
print(is_sorting) # Yes or No
if is_sorting:
# unused_cmp: 使われない比較器かどうか (is_sorting=True の場合のみ)
unused_cmp = is_sorting.get_data()
assert len(unused_cmp) == m
print(sum(unused_cmp))
print(*map(lambda e: e[0] + 1, filter(lambda e: e[1], enumerate(unused_cmp))))
else:
# unsorted_pos: ソートされない可能性のある位置 (is_sorting=False の場合のみ)
unsorted_pos = is_sorting.get_error()
assert len(unsorted_pos) == n - 1
print(sum(unsorted_pos))
print(*map(lambda e: e[0] + 1, filter(lambda e: e[1], enumerate(unsorted_pos))))
main()