結果

問題 No.117 組み合わせの数
ユーザー eitahoeitaho
提出日時 2015-01-31 18:03:21
言語 C#(csc)
(csc 3.9.0)
結果
AC  
実行時間 584 ms / 5,000 ms
コード長 3,349 bytes
コンパイル時間 1,055 ms
コンパイル使用メモリ 119,508 KB
実行使用メモリ 61,592 KB
最終ジャッジ日時 2024-06-23 04:35:43
合計ジャッジ時間 2,972 ms
ジャッジサーバーID
(参考情報)
judge3 / judge4
このコードへのチャレンジ
(要ログイン)

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 584 ms
61,592 KB
権限があれば一括ダウンロードができます
コンパイルメッセージ
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));
        Console.ReadLine();
    }

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

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


    static long ModPower(long x, long n, long mod) // x ^ n
    {
        long res = 1;
        while (n > 0)
        {
            if ((n & 1) == 1) { res = (res * x) % mod; }
            x = (x * x) % mod;
            n >>= 1;
        }
        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