結果

問題 No.5025 Forest Fire
コンテスト
ユーザー Koi
提出日時 2026-10-24 15:33:50
言語 PyPy3
(8.0.0 + ACL)
コンパイル:
pypy3 -mpy_compile _filename_
実行:
pypy3 _filename_
結果
AC  
実行時間 1,438 ms / 2,000 ms
+ 461µs
コード長 1,378 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 79 ms
コンパイル使用メモリ 64,448 KB
実行使用メモリ 93,864 KB
スコア 154,069,626
最終ジャッジ日時 2026-10-24 15:35:55
合計ジャッジ時間 124,091 ms
ジャッジサーバーID
（参考情報） 		judge4_0 / <nil>
純コード判定しない問題か言語
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 50
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

from atcoder import dsu
N, K = map(int, input().split())
D = []
for i in range(N):
    S = input()
    L = []
    for j in range(N):
        L.append(int(S[j]))
    D.append(L)
dx = [0, 1, 0, -1]
dy = [1, 0, -1, 0]

def merge(i, j):
    return i * N + j
def sep(ij):
    return ij //N, ij % N
def calc_score(L):
    S = set(L)
    DS = dsu.DSU(N * N)
    for i in range(N):
        for j in range(N):
            if((i, j) in S or D[i][j] < 5):
                continue
            ij = merge(i, j)
            for k in range(4):
                x2 = i + dx[k]
                y2 = j + dy[k]
                if(0 <= x2 < N and 0 <= y2 < N):
                    if((x2, y2) not in S and D[x2][y2] >= 5):
                        xy2 = merge(x2, y2)
                        DS.merge(ij, xy2)
    s = 0
    # print(DS.groups())
    for l in DS.groups():
        s += len(l) ** 2
    return s
                    
                
best_ans = []
best_score = calc_score([])
for i in range(N):
    for j in range(N):
        L = set()
        for x in range(N):
            if(D[x][i] != 0):
                L.add((x, i))
        for y in range(N):
            if(D[j][y] != 0):
                L.add((j, y))
        score = calc_score(L)
        if(score < best_score):
            best_score = score
            best_ans = L
print(len(best_ans))
for x, y in best_ans:
    print(x, y)
0