結果

問題 No.1364 [Renaming] Road to Cherry from Zelkova
コンテスト
ユーザー kakel-san
提出日時 2024-08-08 23:53:07
言語 C#
(.NET 8.0.404)
結果
WA  
実行時間 -
コード長 3,298 bytes
コンパイル時間 9,865 ms
コンパイル使用メモリ 167,168 KB
実行使用メモリ 187,328 KB
最終ジャッジ日時 2024-08-08 23:53:37
合計ジャッジ時間 26,618 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 20 WA * 25
権限があれば一括ダウンロードができます
コンパイルメッセージ
  復元対象のプロジェクトを決定しています...
  /home/judge/data/code/main.csproj を復元しました (94 ms)。
MSBuild のバージョン 17.9.6+a4ecab324 (.NET)
  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();
    public static void Main()
    {
        Solve();
    }
    static void Solve()
    {
        var c = NList;
        var (n, m) = (c[0], c[1]);
        ++n;
        var map = NArr(m);
        var tree = new List<(int to, long len, long count)>[n];
        for (var i = 0; i < tree.Length; ++i) tree[i] = new List<(int to, long len, long count)>();
        var revtree = new List<int>[n];
        for (var i = 0; i < revtree.Length; ++i) revtree[i] = new List<int>();
        var refs = new int[n];
        foreach (var edge in map)
        {
            tree[edge[0]].Add((edge[1], edge[2], edge[3]));
            revtree[edge[1]].Add(edge[0]);
            ++refs[edge[1]];
        }
        var visited = new bool[n];
        var lengths = new long[n];
        var counts = new long[n];
        counts[0] = 1;
        var mod = 1_000_000_007;
        var q = new Queue<int>();
        for (var i = 0; i < n; ++i) if (refs[i] == 0)
        {
            visited[i] = true;
            q.Enqueue(i);
        }
        while (q.Count > 0)
        {
            var cur = q.Dequeue();
            foreach (var next in tree[cur])
            {
                counts[next.to] = (counts[next.to] + counts[cur] * next.count % mod) % mod;
                lengths[next.to] = (lengths[next.to] + counts[cur] * next.len % mod * next.count % mod +
                    lengths[cur] * Exp(counts[cur], mod - 2, mod) % mod * next.count % mod) % mod;
                --refs[next.to];
                if (refs[next.to] == 0)
                {
                    visited[next.to] = true;
                    q.Enqueue(next.to);
                }
            }
        }
        var reachable0 = new bool[n];
        DFS(0, tree, reachable0);
        var reachableN = new bool[n];
        DFS(n - 1, revtree, reachableN);
        for (var i = 0; i < n; ++i) if (reachable0[i] && reachableN[i] && !visited[i])
        {
            WriteLine("INF");
            return;
        }
        WriteLine(lengths[^1]);
    }
    static void DFS(int cur, List<(int to, long len, long count)>[] tree, bool[] visited)
    {
        visited[cur] = true;
        foreach (var next in tree[cur])
        {
            if (visited[next.to]) continue;
            DFS(next.to, tree, visited);
        }
    }
    static void DFS(int cur, List<int>[] tree, bool[] visited)
    {
        visited[cur] = true;
        foreach (var next in tree[cur])
        {
            if (visited[next]) continue;
            DFS(next, tree, visited);
        }
    }
    static long Exp(long n, long p, int mod)
    {
        long _n = n % mod;
        var _p = p;
        var result = 1L;
        if ((_p & 1) == 1) result *= _n;
        while (_p > 0)
        {
            _n = _n * _n % mod;
            _p >>= 1;
            if ((_p & 1) == 1) result = result * _n % mod;
        }
        return result;
    }
}
0