結果

問題 No.3201 Corporate Synergy
ユーザー kakel-san
提出日時 2025-07-11 23:41:09
言語 C#
(.NET 8.0.404)
結果
AC  
実行時間 70 ms / 2,000 ms
コード長 6,750 bytes
コンパイル時間 7,749 ms
コンパイル使用メモリ 166,868 KB
実行使用メモリ 189,220 KB
最終ジャッジ日時 2025-07-11 23:41:21
合計ジャッジ時間 10,390 ms
ジャッジサーバーID
(参考情報)
judge5 / judge4
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 2
other AC * 20
権限があれば一括ダウンロードができます
コンパイルメッセージ
  復元対象のプロジェクトを決定しています...
  /home/judge/data/code/main.csproj を復元しました (97 ミリ秒)。
  main -> /home/judge/data/code/bin/Release/net8.0/main.dll
  main -> /home/judge/data/code/bin/Release/net8.0/publish/

ソースコード

diff #

using System;
using static System.Console;
using System.Linq;
using System.Collections.Generic;

class Program
{
    static int NN => int.Parse(ReadLine());
    static int[] NList => ReadLine().Split().Select(int.Parse).ToArray();
    static int[][] NArr(long n) => Enumerable.Repeat(0, (int)n).Select(_ => NList).ToArray();
    static int[] LList(long n) => Enumerable.Repeat(0, (int)n).Select(_ => int.Parse(ReadLine())).ToArray();
    public static void Main()
    {
        Solve();
    }
    static void Solve()
    {
        var n = NN;
        var p = NList;
        var m = NN;
        var map = NArr(m);
        var k = NN;
        var syn = NArr(k);
        var f = new MaxFlowGraph(n + k + 2);
        var s = n + k;
        var t = s + 1;
        var take = 0L;
        for (var i = 0; i < n; ++i)
        {
            if (p[i] > 0)
            {
                take += p[i];
                f.AddEdge(s, i, p[i]);
                f.AddEdge(i, t, 0);
            }
            else
            {
                f.AddEdge(s, i, 0);
                f.AddEdge(i, t, -p[i]);
            }
        }
        var INF = long.MaxValue / 2;
        for (var i = 0; i < m; ++i)
        {
            f.AddEdge(map[i][1] - 1, map[i][0] - 1, INF);
        }
        for (var i = 0; i < k; ++i)
        {
            take += syn[i][2];
            f.AddEdge(s, n + i, syn[i][2]);
            f.AddEdge(n + i, t, 0);
            f.AddEdge(n + i, syn[i][0] - 1, INF);
            f.AddEdge(n + i, syn[i][1] - 1, INF);
        }
        WriteLine(take - f.Flow(s, t));
    }
    class MaxFlowGraph
    {
        int _n;
        long max = long.MaxValue;
        class EdgeI
        {
            public int to { get; private set; }
            public int rev { get; private set; }
            public long cap { get; set; }
            public EdgeI(int to, int rev, long cap)
            {
                this.to = to;
                this.rev = rev;
                this.cap = cap;
            }
        }
        List<KeyValuePair<int, int>> pos = new List<KeyValuePair<int, int>>();
        List<EdgeI>[] g;

        public MaxFlowGraph(int n)
        {
            _n = n;
            g = new List<EdgeI>[n];
            for (var i = 0; i < n; ++i) g[i] = new List<EdgeI>();
        }
        public int AddEdge(int from, int to, long cap)
        {
            var m = pos.Count;
            pos.Add(new KeyValuePair<int, int>(from, g[from].Count));
            g[from].Add(new EdgeI(to, g[to].Count, cap));
            g[to].Add(new EdgeI(from, g[from].Count - 1, 0));
            return m;
        }
        public class Edge
        {
            public int from { get; private set; }
            public int to { get; private set; }
            public long cap { get; private set; }
            public long flow { get; private set; }
            public Edge(int from, int to, long cap, long flow)
            {
                this.from = from;
                this.to = to;
                this.cap = cap;
                this.flow = flow;
            }
        }
        public Edge GetEdge(int i)
        {
            var m = pos.Count;
            var e = g[pos[i].Key][pos[i].Value];
            var re = g[e.to][e.rev];
            return new Edge(pos[i].Key, e.to, e.cap + re.cap, re.cap);
        }
        public List<Edge> GetEdges()
        {
            var m = pos.Count;
            var result = new List<Edge>();
            for (var i = 0; i < m; ++i) result.Add(GetEdge(i));
            return result;
        }
        public void ChangeEdge(int i, long new_cap, long new_flow)
        {
            var m = pos.Count;
            var e = g[pos[i].Key][pos[i].Value];
            var re = g[e.to][e.rev];
            e.cap = new_cap - new_flow;
            re.cap = new_flow;
        }
        public long Flow(int s, int t)
        {
            return Flow(s, t, max);
        }
        public long Flow(int s, int t, long limit)
        {
            var level = new int[_n];
            var iter = new int[_n];
            var que = new Queue<int>();
            void bfs()
            {
                for (var i = 0; i < _n; ++i) level[i] = -1;
                level[s] = 0;
                que.Clear();
                que.Enqueue(s);
                while (que.Count > 0)
                {
                    var v = que.Dequeue();
                    foreach (var e in g[v])
                    {
                        if (e.cap == 0 || level[e.to] >= 0) continue;
                        level[e.to] = level[v] + 1;
                        if (e.to == t) return;
                        que.Enqueue(e.to);
                    }
                }
            }
            long dfs(int v, long up)
            {
                if (v == s) return up;
                long res = 0;
                var level_v = level[v];
                for (var i = iter[v]; i < g[v].Count; ++i)
                {
                    var e = g[v][i];
                    if (level_v <= level[e.to] || g[e.to][e.rev].cap == 0) continue;
                    long d = dfs(e.to, Min(up - res, g[e.to][e.rev].cap));
                    if (d <= 0) continue;
                    g[v][i].cap += d;
                    g[e.to][e.rev].cap -= d;
                    res += d;
                    if (res == up) break;
                }
                return res;
            }
            long flow = 0;
            while (flow < limit)
            {
                bfs();
                if (level[t] == -1) break;
                for (var i = 0; i < iter.Length; ++i) iter[i] = 0;
                while (flow < limit)
                {
                    var f = dfs(t, limit - flow);
                    if (f == 0) break;
                    flow += f;
                }
            }
            return flow;
        }
        long Min(long a, long b)
        {
            return a.CompareTo(b) < 0 ? a : b;
        }
        public bool[] MinCut(int s)
        {
            var visited = new bool[_n];
            var que = new Queue<int>();
            que.Enqueue(s);
            while (que.Count > 0)
            {
                var p = que.Dequeue();
                visited[p] = true;
                foreach (var e in g[p])
                {
                    if (e.cap != 0 && !visited[e.to])
                    {
                        visited[e.to] = true;
                        que.Enqueue(e.to);
                    }
                }
            }
            return visited;
        }
    }
}
0