#410558 (C#(csc)) No.957 植林

提出ソース

結果

問題	No.957 植林
ユーザー	りあん
提出日時	2019-12-19 23:38:49
言語	C#(csc) (csc 3.9.0)
結果	TLE
実行時間	-
コード長	9,743 bytes
コンパイル時間	1,168 ms
コンパイル使用メモリ	124,148 KB
実行使用メモリ	103,236 KB
最終ジャッジ日時	2024-07-07 02:08:22
合計ジャッジ時間	11,958 ms
ジャッジサーバーID （参考情報）	judge1 / judge4

このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示

入力	結果	実行時間実行使用メモリ
testcase_00	AC	33 ms 32,360 KB
testcase_01	AC	33 ms 25,676 KB
testcase_02	AC	33 ms 25,460 KB
testcase_03	AC	604 ms 84,368 KB
testcase_04	AC	500 ms 80,700 KB
testcase_05	AC	571 ms 90,848 KB
testcase_06	AC	639 ms 98,032 KB
testcase_07	AC	654 ms 89,620 KB
testcase_08	AC	334 ms 84,256 KB
testcase_09	AC	302 ms 84,748 KB
testcase_10	AC	398 ms 92,684 KB
testcase_11	AC	377 ms 90,328 KB
testcase_12	AC	399 ms 87,752 KB
testcase_13	AC	269 ms 77,072 KB
testcase_14	AC	319 ms 85,080 KB
testcase_15	AC	312 ms 82,160 KB
testcase_16	AC	264 ms 77,256 KB
testcase_17	AC	264 ms 75,188 KB
testcase_18	TLE	-
testcase_19	--	-
testcase_20	--	-
testcase_21	--	-
testcase_22	--	-
testcase_23	--	-
testcase_24	--	-
testcase_25	--	-
testcase_26	--	-
testcase_27	--	-
testcase_28	--	-
testcase_29	--	-
testcase_30	--	-
testcase_31	--	-
testcase_32	--	-
testcase_33	--	-
testcase_34	--	-
testcase_35	--	-
testcase_36	--	-
testcase_37	--	-
testcase_38	--	-
testcase_39	--	-
testcase_40	--	-
testcase_41	--	-
testcase_42	--	-
testcase_43	--	-
testcase_44	--	-
testcase_45	--	-
testcase_46	--	-
testcase_47	--	-

権限があれば一括ダウンロードができます

コンパイルメッセージ

Microsoft (R) Visual C# Compiler version 3.9.0-6.21124.20 (db94f4cc)
Copyright (C) Microsoft Corporation. All rights reserved.

ソースコード

raw source code

using System;
using System.Collections.Generic;
using System.Linq;
using System.IO;
using System.Threading;
using System.Text;
using System.Diagnostics;
using static util;
using P = pair<int, int>;

class Program {
    static void Main(string[] args) {
        var sw = new StreamWriter(Console.OpenStandardOutput()) { AutoFlush = false };
        var solver = new Solver(sw);
        // var t = new Thread(solver.solve, 1 << 26); // 64 MB
        // t.Start();
        // t.Join();
        solver.solve();
        sw.Flush();
    }
}

class Solver {
    StreamWriter sw;
    Scan sc;
    void Prt(string a) => sw.WriteLine(a);
    void Prt<T>(IEnumerable<T> a) => Prt(string.Join(" ", a));
    void Prt(params object[] a) => Prt(string.Join(" ", a));
    public Solver(StreamWriter sw) {
        this.sw = sw;
        this.sc = new Scan();
    }

    public void solve() {
        int h, w;
        // h = w = 300;
        sc.Multi(out h, out w);
        var g = new long[h][];
        // var rnd = new Random();
        for (int i = 0; i < h; i++)
        {
            // g[i] = new long[w];
            // for (int j = 0; j < w; j++)
            // {
            //     g[i][j] = rnd.Next(1000000000);
            // }
            g[i] = sc.LongArr;
        }
        // var r = new long[h];
        // for (int i = 0; i < h; i++)
        // {
        //     r[i] = rnd.Next(1000000000);
        // }
        // var c = new long[w];
        // for (int i = 0; i < w; i++)
        // {
        //     c[i] = rnd.Next(1000000000);
        // }
        var r = sc.LongArr;
        var c = sc.LongArr;
        int N = h + w + h * w + 2;
        int S = N - 2;
        int T = N - 1;
        var mf = new MaxFlow(N);
        var lis = new List<pair<P, long>>();
        for (int i = 0; i < h; i++)
        {
            lis.Add(new pair<P, long>(new P(S, i), r[i]));
            // mf.add_edge(S, i, r[i]);
        }
        for (int i = 0; i < w; i++)
        {
            lis.Add(new pair<P, long>(new P(S, h + i), c[i]));
            // mf.add_edge(S, h + i, c[i]);
        }
        long inf = (long)1e14;
        for (int i = 0; i < h; i++)
        {
            for (int j = 0; j < w; j++)
            {
                lis.Add(new pair<P, long>(new P(i, h + w + i * w + j), inf));
                // mf.add_edge(i, h + w + i * w + j, inf);
                lis.Add(new pair<P, long>(new P(h + j, h + w + i * w + j), inf));
                // mf.add_edge(h + j, h + w + i * w + j, inf);
                lis.Add(new pair<P, long>(new P(h + w + i * w + j, T), g[i][j]));
                // mf.add_edge(h + w + i * w + j, T, g[i][j]);
            }
        }
        // lis.shuffle(new Random());
        foreach (var item in lis)
        {
            mf.add_edge(item.v1.v1, item.v1.v2, item.v2);
        }
        Prt(r.Sum() + c.Sum() - mf.run(S, T));
    }
}

class pair<T, U> : IComparable<pair<T, U>> {
    public T v1;
    public U v2;
    public pair() : this(default(T), default(U)) {}
    public pair(T v1, U v2) { this.v1 = v1; this.v2 = v2; }
    public int CompareTo(pair<T, U> a) {
        int c = Comparer<T>.Default.Compare(v1, a.v1);
        return c != 0 ? c : Comparer<U>.Default.Compare(v2, a.v2);
    }
    public override string ToString() => v1 + " " + v2;
    public void Deconstruct(out T a, out U b) { a = v1; b = v2; }
}
static class util {
    public static readonly int M = 1000000007;
    // public static readonly int M = 998244353;
    public static readonly long LM = 1L << 60;
    public static readonly double eps = 1e-11;
    public static void DBG(string a) => Console.Error.WriteLine(a);
    public static void DBG<T>(IEnumerable<T> a) => DBG(string.Join(" ", a));
    public static void DBG(params object[] a) => DBG(string.Join(" ", a));
    public static void Assert(bool cond) { if (!cond) throw new Exception(); }
    public static pair<T, U> make_pair<T, U>(T v1, U v2) => new pair<T, U>(v1, v2);
    public static int CompareList<T>(IList<T> a, IList<T> b) where T : IComparable<T> {
        for (int i = 0; i < a.Count && i < b.Count; i++)
            if (a[i].CompareTo(b[i]) != 0) return a[i].CompareTo(b[i]);
        return a.Count.CompareTo(b.Count);
    }
    public static bool inside(int i, int j, int h, int w) => i >= 0 && i < h && j >= 0 && j < w;
    static readonly int[] dd = { 0, 1, 0, -1 };
    // static readonly string dstring = "RDLU";
    public static P[] adjacents(int i, int j)
        => Enumerable.Range(0, dd.Length).Select(k => new P(i + dd[k], j + dd[k ^ 1])).ToArray();
    public static P[] adjacents(int i, int j, int h, int w)
        => Enumerable.Range(0, dd.Length).Select(k => new P(i + dd[k], j + dd[k ^ 1]))
                                         .Where(p => inside(p.v1, p.v2, h, w)).ToArray();
    public static P[] adjacents(this P p) => adjacents(p.v1, p.v2);
    public static P[] adjacents(this P p, int h, int w) => adjacents(p.v1, p.v2, h, w);
    public static Dictionary<T, int> compress<T>(this IEnumerable<T> a)
        => a.Distinct().OrderBy(v => v).Select((v, i) => new { v, i }).ToDictionary(p => p.v, p => p.i);
    public static Dictionary<T, int> compress<T>(params IEnumerable<T>[] a) => compress(a.Aggregate(Enumerable.Union));
    public static T[] inv<T>(this Dictionary<T, int> dic) {
        var res = new T[dic.Count];
        foreach (var item in dic) res[item.Value] = item.Key;
        return res;
    }
    public static void shuffle<T>(this IList<T> a, Random rnd) {
        for (int i = a.Count - 1; i > 0 ; i--) {
            int j = rnd.Next(i + 1);
            swap(a, j, i);
        }
    }
    public static void swap<T>(ref T a, ref T b) { var t = a; a = b; b = t; }
    public static void swap<T>(this IList<T> a, int i, int j) { var t = a[i]; a[i] = a[j]; a[j] = t; }
    public static T[] copy<T>(this IList<T> a) {
        var ret = new T[a.Count];
        for (int i = 0; i < a.Count; i++) ret[i] = a[i];
        return ret;
    }
}

class Scan {
    StreamReader sr;
    public Scan() { sr = new StreamReader(Console.OpenStandardInput()); }
    public Scan(string path) { sr = new StreamReader(path); }
    public int Int => int.Parse(Str);
    public long Long => long.Parse(Str);
    public double Double => double.Parse(Str);
    public string Str => sr.ReadLine().Trim();
    public pair<T, U> Pair<T, U>() {
        T a; U b;
        Multi(out a, out b);
        return new pair<T, U>(a, b);
    }
    public P P => Pair<int, int>();
    public int[] IntArr => StrArr.Select(int.Parse).ToArray();
    public long[] LongArr => StrArr.Select(long.Parse).ToArray();
    public double[] DoubleArr => StrArr.Select(double.Parse).ToArray();
    public string[] StrArr => Str.Split(new[]{' '}, StringSplitOptions.RemoveEmptyEntries);
    bool eq<T, U>() => typeof(T).Equals(typeof(U));
    T ct<T, U>(U a) => (T)Convert.ChangeType(a, typeof(T));
    T cv<T>(string s) => eq<T, int>()    ? ct<T, int>(int.Parse(s))
                       : eq<T, long>()   ? ct<T, long>(long.Parse(s))
                       : eq<T, double>() ? ct<T, double>(double.Parse(s))
                       : eq<T, char>()   ? ct<T, char>(s[0])
                                         : ct<T, string>(s);
    public void Multi<T>(out T a) => a = cv<T>(Str);
    public void Multi<T, U>(out T a, out U b)
    { var ar = StrArr; a = cv<T>(ar[0]); b = cv<U>(ar[1]); }
    public void Multi<T, U, V>(out T a, out U b, out V c)
    { var ar = StrArr; a = cv<T>(ar[0]); b = cv<U>(ar[1]); c = cv<V>(ar[2]); }
    public void Multi<T, U, V, W>(out T a, out U b, out V c, out W d)
    { var ar = StrArr; a = cv<T>(ar[0]); b = cv<U>(ar[1]); c = cv<V>(ar[2]); d = cv<W>(ar[3]); }
    public void Multi<T, U, V, W, X>(out T a, out U b, out V c, out W d, out X e)
    { var ar = StrArr; a = cv<T>(ar[0]); b = cv<U>(ar[1]); c = cv<V>(ar[2]); d = cv<W>(ar[3]); e = cv<X>(ar[4]); }
}

class MaxFlow {
    public class edge {
        public int to;
        public long cap;
        public int rev;
        public edge(int t, long c, int r) { to = t; cap = c; rev = r; }
    }
    int V;
    List<edge>[] G;
    int[] itr, lev;
    public List<edge>[] getEdges() => G;

    public MaxFlow(int v) {
        V = v; G = new List<edge>[v];
        for (int i = 0; i < v; ++i) G[i] = new List<edge>();
    }
    public void add_edge(int frm, int to, long c) => add_edge(frm, to, c, true);
    public void add_edge(int frm, int to, long c, bool dir) {
        G[frm].Add(new edge(to, c, G[to].Count));
        G[to].Add(new edge(frm, dir ? 0 : c, G[frm].Count - 1));
    }
    void bfs(int s) {
        lev = new int[V];
        for (int i = 0; i < V; ++i) lev[i] = -1;
        var q = new Queue<int>();
        lev[s] = 0;
        q.Enqueue(s);
        while (q.Count > 0) {
            int v = q.Dequeue();
            foreach (var e in G[v]) {
                if (e.cap > 0 && lev[e.to] < 0) {
                    lev[e.to] = lev[v] + 1;
                    q.Enqueue(e.to);
                }
            }
        }
    }
    long dfs(int v, int t, long f) {
        if (v == t) return f;
        for (; itr[v] < G[v].Count; ++itr[v]) {
            var e = G[v][itr[v]];
            if (e.cap > 0 && lev[v] < lev[e.to]) {
                long d = dfs(e.to, t, f < e.cap ? f : e.cap);
                if (d > 0) {
                    e.cap -= d;
                    G[e.to][e.rev].cap += d;
                    return d;
                }
            }
        }
        return 0;
    }

    public long run(int s, int t) {
        long ret = 0;
        bfs(s);
        while (lev[t] >= 0) {
            itr = new int[V];
            long f;
            while ((f = dfs(s, t, util.LM)) > 0) ret += f;
            bfs(s);
        }
        return ret;
    }
}

yukicoder

結果

テストケース

コンパイルメッセージ

ソースコード