結果

問題 No.2561 みんな大好きmod 998
ユーザー kakel-san
提出日時 2023-12-02 14:46:05
言語 C#
(.NET 8.0.404)
結果
AC  
実行時間 292 ms / 4,000 ms
コード長 1,571 bytes
コンパイル時間 16,085 ms
コンパイル使用メモリ 167,836 KB
実行使用メモリ 184,704 KB
最終ジャッジ日時 2024-09-26 17:20:04
合計ジャッジ時間 17,128 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 44
権限があれば一括ダウンロードができます
コンパイルメッセージ
  復元対象のプロジェクトを決定しています...
  /home/judge/data/code/main.csproj を復元しました (261 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, k) = (c[0], c[1]);
        var a = NList;
        var ans = 0;
        foreach (var ncr in NCRList(n, k))
        {
            var s1 = 0;
            var s2 = 0;
            foreach (var ni in ncr)
            {
                s1 = (s1 + a[ni]) % 998;
                s2 = (s2 + a[ni]) % 998244353;
            }
            if (s1 >= s2)
            {
                ++ans;
            }
        }
        WriteLine(ans % 998);
    }
    static IEnumerable<int[]> NCRList(int n, int r)
    {
        var ans = new int[r];
        for (var i = 0; i < ans.Length; ++i) ans[i] = i;
        ans[ans.Length - 1] = ans.Length - 2;
        var pos = r - 1;
        while (pos >= 0)
        {
            ++ans[pos];
            while (pos < r - 1)
            {
                ++pos;
                ans[pos] = ans[pos - 1] + 1;
            }
            if (pos == r - 1)
            {
                yield return ans;
            }
            while (pos >= 0 && ans[pos] == n + pos - r)
            {
                --pos;
            }
        }
    }
}
0