結果

問題 No.3040 Aoiスコア
ユーザー tobisatis
提出日時 2025-02-28 21:59:13
言語 C#
(.NET 8.0.404)
結果
AC  
実行時間 160 ms / 1,000 ms
コード長 4,808 bytes
コンパイル時間 7,722 ms
コンパイル使用メモリ 170,852 KB
実行使用メモリ 228,444 KB
最終ジャッジ日時 2025-06-20 20:57:48
合計ジャッジ時間 11,036 ms
ジャッジサーバーID
(参考情報)
judge2 / judge5
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 26
権限があれば一括ダウンロードができます
コンパイルメッセージ
  復元対象のプロジェクトを決定しています...
  /home/judge/data/code/main.csproj を復元しました (104 ミリ秒)。
  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 s = String().AsSpan();
    var g = new bool[n, n];
    for (var i = 0; i < m; i++)
    {
        var u = Int() - 1;
        var v = Int() - 1;
        g[u, v] = true;
        g[v, u] = true;
    }
    var tq = 0;
    foreach (var c in s) if (c == '?') tq++;
    var p26 = new ModInt[n + 1];
    p26[0] = 1;
    for (var i = 1; i <= n; i++) p26[i] = p26[i - 1] * 26;
    ModInt ans = 0;
    for (var u = 0; u < n; u++)
    {
        var uc = s[u];
        if (uc != 'o' && uc != '?') continue;
        var uq = tq;
        if (uc == '?') uq--;
        ModInt ca = 0;
        ModInt ci = 0;
        var cq = 0;
        for (var v = 0; v < n; v++) if (g[u, v])
        {
            var vc = s[v];
            if (vc == 'a') ca += 1;
            if (vc == 'i') ci += 1;
            if (vc == '?') cq += 1;
        }
        ans += ca * ci * p26[uq];
        if (cq > 0) ans += (ca + ci) * cq * p26[uq - 1];
        if (cq > 1) ans += cq * (cq - 1) * p26[uq - 2];
    }
    Out(ans);
}

#region
AtCoderIO _io_;
var _backend_ = new StandardIOBackend();
_io_ = new(){ Backend = _backend_ };
Run();
_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; }

    Memory<string> _input = Array.Empty<string>();
    int _iter = 0;
    public string Next()
    {
        while (_iter >= _input.Length) (_input, _iter) = (Backend.ReadLine().Trim().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();
}

readonly record struct ModInt // :
    // IEqualityOperators<ModInt, ModInt, bool>,
    // IAdditiveGroup<ModInt>,
    // IMultiplicativeGroup<ModInt>
{
    public const int Mod = 998244353;
    int V { get; init; }
    public ModInt(long value)
    {
        var v = value % Mod;
        if (v < 0) v += Mod;
        V = (int)v;
    }
    static ModInt New(int value) => new(){ V = value };

    public static implicit operator ModInt(long v) => new(v);
    public static implicit operator int(ModInt modInt) => modInt.V;

    public static ModInt AdditiveIdentity => New(0);
    public static ModInt operator +(ModInt a, ModInt b)
    {
        var v = a.V + b.V;
        if (v >= Mod) v -= Mod;
        return New(v);
    }
    public ModInt AdditiveInverse()
    {
        if (V == 0) return AdditiveIdentity;
        return New(Mod - V);
    }
    public static ModInt operator -(ModInt a, ModInt b)
    {
        var v = a.V - b.V;
        if (v < 0) v += Mod;
        return New(v);
    }

    public static ModInt MultiplicativeIdentity => New(1);
    public static ModInt operator *(ModInt a, ModInt b) => New((int)((long)a.V * b.V % Mod));
    public ModInt MultiplicativeInverse()
    {
        if (V == 0) throw new DivideByZeroException();
        var (d, x, _) = ExtendedGCD(V, Mod);
        if (d > 1) throw new DivideByZeroException();
        return x;
    }
    public static ModInt operator /(ModInt a, ModInt b) => a * b.MultiplicativeInverse();

    static long Power(long v, ulong p, long mod)
    {
        var (res, k) = (1L, v);
        while (p > 0)
        {
            if ((p & 1) > 0) res = res * k % mod;
            k = k * k % mod;
            p >>= 1;
        }
        return res;
    }
    public ModInt Power(long p) => p < 0 ? (MultiplicativeIdentity / V).Power(-p) : Power(V, (ulong)p, Mod);

    static (long d, long x, long y) ExtendedGCD(long a, long b)
    {
        var (x0, y0, x1, y1) = (1L, 0L, 0L, 1L);
        while (b != 0)
        {
            var q = a / b;
            (a, b) = (b, a - q * b);
            (x0, y0, x1, y1) = (x1, y1, x0 - q * x1, y0 - q * y1);
        }
        return (a, x0, y0);
    }

    public override string ToString() => V.ToString();
}
0