結果

問題 No.3158 Collect Stamps
コンテスト
ユーザー tobisatis
提出日時 2025-05-23 19:14:52
言語 C#
(.NET 8.0.404)
結果
AC  
実行時間 1,432 ms / 2,000 ms
コード長 3,043 bytes
コンパイル時間 17,873 ms
コンパイル使用メモリ 172,540 KB
実行使用メモリ 208,860 KB
最終ジャッジ日時 2025-05-23 19:15:39
合計ジャッジ時間 30,196 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 35
権限があれば一括ダウンロードができます
コンパイルメッセージ
  復元対象のプロジェクトを決定しています...
  /home/judge/data/code/main.csproj を復元しました (95 ミリ秒)。
  main -> /home/judge/data/code/bin/Release/net8.0/main.dll
  main -> /home/judge/data/code/bin/Release/net8.0/publish/

ソースコード

diff # 

#nullable enable

using System.Numerics;

void Run()
{
    var n = Int();
    var m = Int();
    var k = Int();
    var a = new bool[n];
    for (var i = 0; i < k; i++) a[Int() - 1] = true;
    var tz = new int[n, n];
    for (var i = 0; i < n; i++) for (var j = 0; j < n; j++) tz[i, j] = Int();
    var b = n.ToFlag();
    var max = long.MaxValue / 8;
    var dp = new long[n, b];
    for (var i = 0; i < n; i++) for (var j = 0; j < b; j++) dp[i, j] = max;
    var q = new PriorityQueue<(int, int, long), long>();
    for (var i = 0; i < n; i++)
    {
        var j = i.ToFlag();
        dp[i, j] = 0;
        q.Enqueue((i, j, 0), 0);
    }
    var ans = max;
    while (q.Count > 0)
    {
        var (l, s, v) = q.Dequeue();
        if (v > dp[l, s]) continue;
        var c = 0;
        for (var i = 0; i < n; i++) if (s.OfFlag(i)) c++;
        if (c >= m && a[l])
        {
            ans = Math.Min(ans, v);
            continue;
        }
        for (var i = 0; i < n; i++) if (!s.OfFlag(i))
        {
            var next = s.WithFlag(i, true);
            var nv = v + tz[l, i];
            if (dp[i, next] <= nv) continue;
            dp[i, next] = nv;
            q.Enqueue((i, next, nv), nv);
        }
    }
    Out(ans);
}

#region
var _io_ = new AtCoderIO(){ Backend = new() };
Run();
_io_.Backend.Flush();

string String() => _io_.Next();
int Int() => int.Parse(String());
void Out(object? x, string? sep = null) => _io_.Out(x, sep);

class AtCoderIO
{
    public required StandardIOBackend Backend { get; init; }

    ReadOnlyMemory<string> _input = Array.Empty<string>();
    int _iter = 0;
    public string Next()
    {
        while (_iter >= _input.Length) (_input, _iter) = (Backend.ReadLine().Split(' '), 0);
        return _input.Span[_iter++];
    }

    public void Out(object? x, string? separator = null)
    {
        if (x == null) return;
        separator ??= Environment.NewLine;
        if (x is System.Collections.IEnumerable a and not string)
        {
            var objects = a.Cast<object>();
            if (separator == Environment.NewLine && !objects.Any()) return;
            x = string.Join(separator, objects);
        }
        Backend.WriteLine(x);
    }
}

class StandardIOBackend
{
    readonly StreamReader _sr = new(Console.OpenStandardInput());
    readonly StreamWriter _sw = new(Console.OpenStandardOutput()) { AutoFlush = false };
    public string ReadLine() => _sr.ReadLine()!;
    public void WriteLine(object? value) => _sw.WriteLine(value);
    public void Flush() => _sw.Flush();
}
#endregion

static class Extensions
{
    public static T[] Repeat<T>(this int time, Func<T> F) => Enumerable.Range(0, time).Select(_ => F()).ToArray();

    public static int ToFlag(this int i)
    {
        if (i < 0 || 31 <= i) throw new IndexOutOfRangeException();
        return 1 << i;
    }
    public static bool OfFlag(this int value, int i) => (value & ToFlag(i)) > 0;
    public static int WithFlag(this int value, int i, bool f) => f ? (value | ToFlag(i)) : (value & ~ToFlag(i));

}
0