結果
問題 | No.177 制作進行の宮森あおいです! |
ユーザー |
![]() |
提出日時 | 2015-05-28 11:59:35 |
言語 | Python3 (3.13.1 + numpy 2.2.1 + scipy 1.14.1) |
結果 |
AC
|
実行時間 | 38 ms / 2,000 ms |
コード長 | 2,551 bytes |
コンパイル時間 | 134 ms |
コンパイル使用メモリ | 12,800 KB |
実行使用メモリ | 11,648 KB |
最終ジャッジ日時 | 2024-07-06 12:06:28 |
合計ジャッジ時間 | 1,270 ms |
ジャッジサーバーID (参考情報) |
judge4 / judge2 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 13 |
ソースコード
#!/usr/bin/env python# -*- coding: utf-8 -*-class Dinic:class Edge:def __init__(self, to, cap, rev):self.to = toself.cap = capself.rev = revdef __repr__(self):return "(to: {0} cap: {1} rev: {2})".format(self.to, self.cap, self. rev)def __init__(self,V):self.V = Vself.size = [0 for i in range(V)]self.G = [[] for i in range(V)]def add_edge(self, _from, to, cap):self.G[_from].append(self.Edge(to, cap, self.size[to]))self.G[to].append(self.Edge(_from, 0, self.size[_from]))self.size[_from] += 1self.size[to] += 1def bfs(self,s):level = [-1 for i in range(self.V)]level[s] = 0q = []q.append(s)while q != []:v = q.pop(0)for u in self.G[v]:if u.cap > 0 and level[u.to] < 0:level[u.to] = level[v] + 1q.append(u.to)return leveldef dfs(self, v, t, f):if v == t: return fwhile self.iterator[v] < self.size[v]:edge = self.G[v][self.iterator[v]]if edge.cap > 0 and self.level[v] < self.level[edge.to]:d = self.dfs(edge.to, t, f if f < edge.cap else edge.cap)if d > 0:self.G[v][self.iterator[v]].cap -= dself.G[edge.to][edge.rev].cap += dreturn dself.iterator[v] += 1return 0def max_flow(self, s, t):flow = 0while True:self.level = self.bfs(s)if self.level[t] < 0:return flowself.iterator = [0 for i in range(self.V)]while True:f = self.dfs(s, t, float('inf'))if f == 0:breakflow += fW = int(input())N = int(input())J = list(map(int,input().split()))M = int(input())C = list(map(int,input().split()))ff = Dinic(N+M+2)for i in range(1,N+1):ff.add_edge(0,i,J[i-1])for i in range(N+1,N+M+1):ff.add_edge(i,N+M+1,C[i-(N+1)])Q = list(map(int,input().split()))counter = 1for j in range(1,N+1):if Q[0] == 0 or Q[counter] != j:ff.add_edge(j, i, 10**8)elif Q[0] != 0 and Q[counter] == j:ff.add_edge(j, i, 0)counter += 1 if counter < Q[0] else 0if ff.max_flow(0,N+M+1) >= W:print('SHIROBAKO')else:print('BANSAKUTSUKITA')