結果

問題 No.2561 みんな大好きmod 998
ユーザー bluemegane
提出日時 2023-12-02 17:09:53
言語 C#
(.NET 8.0.404)
結果
TLE  
実行時間 -
コード長 1,642 bytes
コンパイル時間 8,136 ms
コンパイル使用メモリ 167,604 KB
実行使用メモリ 208,820 KB
最終ジャッジ日時 2024-09-26 21:07:36
合計ジャッジ時間 25,714 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 37 TLE * 7
権限があれば一括ダウンロードができます
コンパイルメッセージ
  復元対象のプロジェクトを決定しています...
  /home/judge/data/code/main.csproj を復元しました (86 ms)。
MSBuild のバージョン 17.9.6+a4ecab324 (.NET)
/home/judge/data/code/Main.cs(5,21): warning CS8981: 型名 'combi' には、小文字の ASCII 文字のみが含まれています。このような名前は、プログラミング言語用に予約されている可能性があります。 [/home/judge/data/code/main.csproj]
  main -> /home/judge/data/code/bin/Release/net8.0/main.dll
  main -> /home/judge/data/code/bin/Release/net8.0/publish/

ソースコード

diff # 

using System.Collections.Generic;
using System.Linq;
using System;

public static class combi
{
    public static IEnumerable<IEnumerable<T>> Comb<T>(this IEnumerable<T> items, int r)
    {
        if (r == 0)
        {
            yield return Enumerable.Empty<T>();
        }
        else
        {
            var i = 1;
            foreach (var x in items)
            {
                var xs = items.Skip(i);
                foreach (var c in Comb(xs, r - 1))
                    yield return c.Before(x);
                i++;
            }
        }
    }
    public static IEnumerable<T> Before<T>(this IEnumerable<T> items, T first)
    {
        yield return first;
        foreach (var i in items)
            yield return i;
    }
}

public class Hello
{
    public static int MOD = 998244353;
    static void Main()
    {
        string[] line = Console.ReadLine().Trim().Split(' ');
        var n = int.Parse(line[0]);
        var k = int.Parse(line[1]);
        line = Console.ReadLine().Trim().Split(' ');
        var a = Array.ConvertAll(line, int.Parse);
        getAns(n, k, a);
    }
    static bool calc(int[] a, int[] p)
    {
        var s998 = 0L;
        var sMOD = 0L;
        foreach (var x in p)
        {
            s998 += a[x];
            s998 %= 998;
            sMOD += a[x];
            sMOD %= MOD;
        }
        return sMOD <= s998;
    }
    static void getAns(int n, int k, int[] a)
    {
        var c = 0L;
        foreach (var x in Enumerable.Range(0, n).Comb(k))
        {
            var p = x.ToArray();
            if (calc(a, p)) c++;
        }
        Console.WriteLine(c %= 998);
    }
}
0