結果
問題 | No.1266 7 Colors |
ユーザー |
|
提出日時 | 2021-01-21 15:00:38 |
言語 | PyPy3 (7.3.15) |
結果 |
TLE
|
実行時間 | - |
コード長 | 1,914 bytes |
コンパイル時間 | 249 ms |
コンパイル使用メモリ | 82,048 KB |
実行使用メモリ | 399,428 KB |
最終ジャッジ日時 | 2024-12-24 15:04:40 |
合計ジャッジ時間 | 74,575 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge2 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 2 TLE * 17 |
ソースコード
#region Header#!/usr/bin/env python3# from typing import *import sysimport ioimport mathimport collectionsimport decimalimport itertoolsfrom queue import PriorityQueueimport bisectimport heapqdef input():return sys.stdin.readline()[:-1]sys.setrecursionlimit(1000000)#endregion# _INPUT = """3 2 3# 1010000# 1000000# 0010000# 1 2# 1 3# 2 1 0# 1 1 2# 2 1 0# """# sys.stdin = io.StringIO(_INPUT)def dfs(G, colors, seen, city, color):next_colors = set([color])seen[city][color] = Truei = (color + 1) % 7while colors[city][i] and (i not in next_colors):if not seen[city][i]:seen[city][i] = Truenext_colors.add(i)i = (i + 1) % 7i = (color - 1) % 7while colors[city][i] and (i not in next_colors):if not seen[city][i]:seen[city][i] = Truenext_colors.add(i)i = (i - 1) % 7n = 0for next_color in next_colors:n += 1for next_city in G[city]:if colors[next_city][next_color] and (not seen[next_city][next_color]):n += dfs(G, colors, seen, next_city, next_color)return ndef main():N, M, Q = map(int, input().split())colors = [None for _ in range(N)]for i in range(N):s = input()colors[i] = [(s[j] == '1') for j in range(7)]G = [list() for _ in range(N)]for _ in range(M):_u, _v = map(int, input().split())G[_u-1].append(_v-1)G[_v-1].append(_u-1)for _ in range(Q):t, x, y = map(int, input().split())if t == 1:colors[x-1][y-1] = Trueelse:seen = [[False for _ in range(7)] for _ in range(N)]n = dfs(G, colors, seen, x-1, 0)print(n)if __name__ == '__main__':main()