トップページに戻る
次の競技プログラミングの問題へ
前の競技プログラミングの問題へ
No.16 累乗の加算
■■■問題■■■
Maliaは、与えられた数式を計算しようとしている。
数式は1つの変数xのみで構成されており、各i項はxのai乗で構成されていて、
xのa1乗 + xのa2乗 + ・・・ + xのaN乗
という形になっている。
xの累乗を計算するのは大変だと気づいたあなたは、代わりに計算してあげることになりました。
xと各ai (1 <= i <= n)が与えられた時の計算値を求めてください。
答えの値が非常に大きくなるので計算値に対して1000003で割った余りを求めてください。
計算途中の値が32ビットに収まらないことがあります。
■■■入力■■■
x N
a1 a2 ・・・ aN
1行目に変数値を表す整数値x (1 <= x <= 100)と項の数を表す整数値N (1 <= N <= 100)が
半角スペース区切りで与えられる。
2行目には、各項の指数値を表す整数値 ai (0 <= ai <= 1億)が半角スペース区切りで与えられる。
■■■出力■■■
xのa1乗 + xのa2乗 + ・・・ + xのaN乗 の計算値を1000003で割った余りを求めてください。
C#のソース
using System;
using System.Collections.Generic;
using System.Linq;
class Program
{
static string InputPattern = "Input1";
static List<string> GetInputList()
{
var WillReturn = new List<string>();
if (InputPattern == "Input1") {
WillReturn.Add("2 3");
WillReturn.Add("1 2 3");
//14
//x + xの2乗 + xの3乗 , x=2 を表しております。この計算値はこのようになります。
//x + xの2乗 + xの3乗 = 2 + 4 + 8 = 14
}
else if (InputPattern == "Input2") {
WillReturn.Add("2 2");
WillReturn.Add("0 100");
//253110
//xの0乗 + xの100乗 = 1 + 1267650600228229401496703205376
//となるが、1000003で割った余りは、253110となる。
}
else {
string wkStr;
while ((wkStr = Console.ReadLine()) != null) WillReturn.Add(wkStr);
}
return WillReturn;
}
static void Main()
{
List<string> InputList = GetInputList();
int X = InputList[0].Split(' ').Select(A => int.Parse(A)).First();
int[] AArr = InputList[1].Split(' ').Select(A => int.Parse(A)).ToArray();
long Answer = 0;
foreach (int EachInt in AArr) {
Answer = (Answer + DeriveBekijyou(X, EachInt)) % 1000003;
}
Console.WriteLine(Answer);
}
//繰り返し2乗法で、XのN乗を求める
static long DeriveBekijyou(int pX, int pN)
{
long CurrJyousuu = pX % 1000003;
long CurrShisuu = 1;
long WillReturn = 1;
while (true) {
//対象ビットが立っている場合
if ((pN & CurrShisuu) > 0) {
WillReturn = (WillReturn * CurrJyousuu) % 1000003;
}
CurrShisuu *= 2;
if (CurrShisuu > pN) return WillReturn;
CurrJyousuu = (CurrJyousuu * CurrJyousuu) % 1000003;
}
}
}
解説
繰り返し2乗法のオーダーは、Log2のNなので、
繰り返し2乗法の結果を再利用しなくても、TLEにならないと判断しました。