結果

問題 No.117 組み合わせの数
ユーザー eitaho
提出日時 2015-01-31 20:01:21
言語 C#(csc)
(csc 3.9.0)
結果
AC  
実行時間 594 ms / 5,000 ms
コード長 3,408 bytes
コンパイル時間 1,091 ms
コンパイル使用メモリ 110,080 KB
実行使用メモリ 53,504 KB
最終ジャッジ日時 2024-06-23 04:38:53
合計ジャッジ時間 2,839 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 1
権限があれば一括ダウンロードができます
コンパイルメッセージ
Microsoft (R) Visual C# Compiler version 3.9.0-6.21124.20 (db94f4cc)
Copyright (C) Microsoft Corporation. All rights reserved.

ソースコード

diff # 

using System;
using System.IO;
using System.Text;
using System.Linq;
using System.Diagnostics;
using System.Collections.Generic;
using Enu = System.Linq.Enumerable;
using System.Text.RegularExpressions;

class Program
{
    static readonly int MaxN = (int)1e6;
    static readonly int Mod = (int)1e9 + 7;
    long[] fact = new long[MaxN * 2 + 1];

    void Solve()
    {
        int T = reader.Int();
        var S = reader.StringArray(T);
        var ans = new long[T];
        fact[0] = 1;
        for (long i = 1; i < fact.Length; i++)
            fact[i] = fact[i - 1] * i % Mod;

        for (int i = 0; i < T; i++)
        {
            var match = Regex.Match(S[i], "([0-9]+?),([0-9]+)");
            int N = int.Parse(match.Groups[1] + ""), K = int.Parse(match.Groups[2] + "");
            if (S[i][0] == 'P') { ans[i] = P(N, K); }
            if (S[i][0] == 'C') { ans[i] = C(N, K); }
            if (S[i][0] == 'H') { ans[i] = (N == 0 && K == 0) ? 1 : C(K + N - 1, N - 1); }
        }

        Console.WriteLine(string.Join("\n", ans));
    }

    long P(int N, int K)
    {
        if (N < 0 || K < 0 || K > N) return 0;
        return fact[N] * ModInverse(fact[N - K], Mod) % Mod;
    }

    long C(int N, int K)
    {
        if (N < 0 || K < 0 || K > N) return 0;
        return fact[N] * ModInverse(fact[K], Mod) % Mod * ModInverse(fact[N - K], Mod) % Mod;
    }


    long ModInverse(long N, long mod)
    {
        long x = 0, y = 0;
        ExtGCD(N, mod, ref x, ref y);
        return (x % mod + mod) % mod;
    }
    static long ExtGCD(long a, long b, ref long x, ref long y)
    {
        if (b == 0) { x = 1; y = 0; return a; }
        long res = ExtGCD(b, a % b, ref y, ref x);
        y -= (a / b) * x;
        return res;
    }


    static void Main() { new Program().Solve(); }
    Reader reader = new Reader(Console.In);
    class Reader
    {
        private readonly TextReader reader;
        private readonly char[] separator = { ' ' };
        private readonly StringSplitOptions removeOp = StringSplitOptions.RemoveEmptyEntries;
        private string[] A = new string[0];
        private int i;

        public Reader(TextReader r) { reader = r; }
        public bool HasNext() { return Enqueue(); }
        public string String() { return Dequeue(); }
        public int Int() { return int.Parse(Dequeue()); }
        public long Long() { return long.Parse(Dequeue()); }
        public double Double() { return double.Parse(Dequeue()); }
        public int[] IntLine() { var s = Line(); return s == "" ? new int[0] : Array.ConvertAll(Split(s), int.Parse); }
        public int[] IntArray(int N) { return Enu.Range(0, N).Select(i => Int()).ToArray(); }
        public int[][] IntGrid(int H) { return Enu.Range(0, H).Select(i => IntLine()).ToArray(); }
        public string[] StringArray(int N) { return Enu.Range(0, N).Select(i => Line()).ToArray(); }
        public string Line() { return reader.ReadLine().Trim(); }
        private string[] Split(string s) { return s.Split(separator, removeOp); }
        private bool Enqueue()
        {
            if (i < A.Length) return true;
            string line = reader.ReadLine();
            if (line == null) return false;
            if (line == "") return Enqueue();
            A = Split(line);
            i = 0;
            return true;
        }
        private string Dequeue() { Enqueue(); return A[i++]; }
    }
}
0