AOJ本の読書メモ   AOJ    次のAOJの問題へ    前のAOJの問題へ

DPL_5_A: Balls and Boxes 1


問題へのリンク

ボール入れ方に制限なし箱の中身は1つ以下箱の中身は1つ以上
区別できる区別できる123
区別できない区別できる456
区別できる区別できない789
区別できない区別できない101112


C#のソース

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

class Program
{
    static string InputPattern = "InputX";

    static List<string> GetInputList()
    {
        var WillReturn = new List<string>();

        if (InputPattern == "Input1") {
            WillReturn.Add("2 3");
            //9
        }
        else if (InputPattern == "Input2") {
            WillReturn.Add("10 5");
            //9765625
        }
        else if (InputPattern == "Input3") {
            WillReturn.Add("100 100");
            //424090053
        }
        else {
            string wkStr;
            while ((wkStr = Console.ReadLine()) != null) WillReturn.Add(wkStr);
        }
        return WillReturn;
    }

    const long Hou = 1000000007;

    static void Main()
    {
        List<string> InputList = GetInputList();
        long[] wkArr = InputList[0].Split(' ').Select(pX => long.Parse(pX)).ToArray();

        long N = wkArr[0];
        long K = wkArr[1];

        long Answer = K;

        for (long I = 2; I <= N; I++) {
            Answer *= K;
            Answer %= Hou;
        }
        Console.WriteLine(Answer);
    }
}


解説

ボール1個につき、箱の数だけ設置箇所があるので

ボールをN、箱をKとして、
KのN乗が解になります。