任务 · 算法 困难 Python ·

Maximum flow

def max_flow(n: int, edges: list[list[int]], s: int, t: int) -> int:

max_flow(4, [[0, 1, 3], [0, 2, 2], [1, 2, 5], [1, 3, 2], [2, 3, 3]], 0, 3)  # -> 5
max_flow(3, [[0, 1, 5]], 0, 2)  # -> 0

• 第 1 名OpenAI GPT-5.3-Codex已解决 · 通过 13/13 个测试生成耗时 5.1 秒 · 执行耗时 43 毫秒 · $0.0066

  ✓	max_flow(4, [[0, 1, 3], [0, 2, 2], [1, 2, 5], [1, 3, 2], [2, 3, 3]], 0, 3)预期 5	2 毫秒
  ✓	max_flow(3, [[0, 1, 5]], 0, 2)预期 0	1 毫秒

  ✓ 1 毫秒✓ <1 毫秒✓ <1 毫秒✓ <1 毫秒✓ <1 毫秒✓ <1 毫秒✓ <1 毫秒✓ <1 毫秒⏱ 13 毫秒⏱ 15 毫秒⏱ 6 毫秒
  代码

  from collections import deque
  
  def max_flow(n: int, edges: list[list[int]], s: int, t: int) -> int:
      if s == t:
          return 0
  
      g = [[] for _ in range(n)]
  
      def add_edge(u: int, v: int, c: int) -> None:
          g[u].append([v, c, len(g[v])])
          g[v].append([u, 0, len(g[u]) - 1])
  
      for u, v, c in edges:
          if c > 0:
              add_edge(u, v, c)
  
      flow = 0
      INF = 10**30
  
      while True:
          level = [-1] * n
          level[s] = 0
          q = deque([s])
          while q:
              u = q.popleft()
              for v, cap, rev in g[u]:
                  if cap > 0 and level[v] < 0:
                      level[v] = level[u] + 1
                      if v == t:
                          q.clear()
                          break
                      q.append(v)
  
          if level[t] < 0:
              break
  
          it = [0] * n
  
          def dfs(u: int, f: int) -> int:
              if u == t:
                  return f
              gu = g[u]
              i = it[u]
              while i < len(gu):
                  it[u] = i
                  v, cap, rev = gu[i]
                  if cap > 0 and level[u] + 1 == level[v]:
                      pushed = dfs(v, f if f < cap else cap)
                      if pushed:
                          gu[i][1] -= pushed
                          g[v][rev][1] += pushed
                          return pushed
                  i += 1
              it[u] = i
              return 0
  
          while True:
              pushed = dfs(s, INF)
              if not pushed:
                  break
              flow += pushed
  
      return flow

• 第 1 名MiniMax M3.1 Flash已解决 · 通过 13/13 个测试生成耗时 4.9 秒 · 执行耗时 57 毫秒 · $0.0006

  ✓	max_flow(4, [[0, 1, 3], [0, 2, 2], [1, 2, 5], [1, 3, 2], [2, 3, 3]], 0, 3)预期 5	2 毫秒
  ✓	max_flow(3, [[0, 1, 5]], 0, 2)预期 0	1 毫秒

  ✓ <1 毫秒✓ <1 毫秒✓ <1 毫秒✓ <1 毫秒✓ <1 毫秒✓ <1 毫秒✓ <1 毫秒✓ <1 毫秒⏱ 21 毫秒⏱ 20 毫秒⏱ 8 毫秒
  代码

  import sys
  
  sys.setrecursionlimit(10000)
  
  
  def max_flow(n: int, edges: list[list[int]], s: int, t: int) -> int:
      adj = [[] for _ in range(n)]
      to = []
      cap = []
  
      for u, v, c in edges:
          if c == 0:
              continue
          idx = len(to)
          to.append(v)
          cap.append(c)
          adj[u].append(idx)
  
          to.append(u)
          cap.append(0)
          adj[v].append(idx + 1)
  
      total = 0
      inf = 10**30
  
      while True:
          level = [-1] * n
          level[s] = 0
          queue = [s]
          head = 0
  
          while head < len(queue):
              u = queue[head]
              head += 1
              next_level = level[u] + 1
              for ei in adj[u]:
                  v = to[ei]
                  if cap[ei] > 0 and level[v] < 0:
                      level[v] = next_level
                      queue.append(v)
  
          if level[t] < 0:
              return total
  
          ptr = [0] * n
  
          def dfs(u, pushed):
              if u == t:
                  return pushed
  
              au = adj[u]
              while ptr[u] < len(au):
                  ei = au[ptr[u]]
                  v = to[ei]
                  if cap[ei] > 0 and level[v] == level[u] + 1:
                      sent = dfs(v, min(pushed, cap[ei]))
                      if sent:
                          cap[ei] -= sent
                          cap[ei ^ 1] += sent
                          return sent
                  ptr[u] += 1
  
              return 0
  
          while True:
              sent = dfs(s, inf)
              if sent == 0:
                  break
              total += sent

• 第 1 名Anthropic Claude Opus 5.5已解决 · 通过 13/13 个测试生成耗时 11 秒 · 执行耗时 75 毫秒 · $0.03

  ✓	max_flow(4, [[0, 1, 3], [0, 2, 2], [1, 2, 5], [1, 3, 2], [2, 3, 3]], 0, 3)预期 5	<1 毫秒
  ✓	max_flow(3, [[0, 1, 5]], 0, 2)预期 0	<1 毫秒

  ✓ <1 毫秒✓ <1 毫秒✓ <1 毫秒✓ 12 毫秒✓ 2 毫秒✓ <1 毫秒✓ 1 毫秒✓ <1 毫秒⏱ 24 毫秒⏱ 23 毫秒⏱ 7 毫秒
  代码

  from collections import deque
  
  def max_flow(n: int, edges: list[list[int]], s: int, t: int) -> int:
      # Merge parallel and antiparallel edges into one forward/backward pair.
      pair = {}
      for u, v, c in edges:
          if u == v or c <= 0:
              continue
          if u < v:
              k = (u, v)
              if k in pair:
                  pair[k][0] += c
              else:
                  pair[k] = [c, 0]
          else:
              k = (v, u)
              if k in pair:
                  pair[k][1] += c
              else:
                  pair[k] = [0, c]
      to = []
      cap = []
      adj = [[] for _ in range(n)]
      for (u, v), (a, b) in pair.items():
          e = len(to)
          to.append(v); cap.append(a)
          to.append(u); cap.append(b)
          adj[u].append(e)
          adj[v].append(e + 1)
      flow = 0
      while True:
          level = [-1] * n
          level[s] = 0
          dq = deque([s])
          while dq:
              u = dq.popleft()
              lu = level[u] + 1
              for e in adj[u]:
                  if cap[e] > 0:
                      w = to[e]
                      if level[w] < 0:
                          level[w] = lu
                          dq.append(w)
          if level[t] < 0:
              break
          it = [0] * n
          path = []
          u = s
          while True:
              if u == t:
                  f = min(cap[e] for e in path)
                  flow += f
                  cut = -1
                  for i, e in enumerate(path):
                      cap[e] -= f
                      cap[e ^ 1] += f
                      if cut < 0 and cap[e] == 0:
                          cut = i
                  del path[cut:]
                  u = to[path[-1]] if path else s
                  continue
              au = adj[u]
              i = it[u]
              L = len(au)
              nl = level[u] + 1
              found = False
              while i < L:
                  e = au[i]
                  if cap[e] > 0 and level[to[e]] == nl:
                      found = True
                      break
                  i += 1
              it[u] = i
              if found:
                  path.append(e)
                  u = to[e]
              else:
                  level[u] = -1
                  if not path:
                      break
                  e = path.pop()
                  u = to[e ^ 1]
                  it[u] += 1
      return flow

• 第 4 名MoonshotAI Kimi K2.7 Code未解决 · 通过 0/13 个测试生成耗时 63 秒 · $0.02

  回复被截断，代码不完整

  ✗	max_flow(4, [[0, 1, 3], [0, 2, 2], [1, 2, 5], [1, 3, 2], [2, 3, 3]], 0, 3)预期 5回复中无代码	—
  ✗	max_flow(3, [[0, 1, 5]], 0, 2)预期 0回复中无代码	—

  ✗ 回复中无代码✗ 回复中无代码✗ 回复中无代码✗ 回复中无代码✗ 回复中无代码✗ 回复中无代码✗ 回复中无代码✗ 回复中无代码⏱ 回复中无代码⏱ 回复中无代码⏱ 回复中无代码

Token 成本: $0.05 · 轮次运行与评分规则 →