[map] 神奇的序列 校OJ2480

题目描述

序列a如下：

a[0] = A;

a[1] = B;

a[i] = a[pp] * a[qq]; ( i>=2，pp = 向下取整(i/k1)，qq = 向下取整(i/k2) )

有Q次询问，每次询问输入pos，请输出a[pos]%mod。

输入

第一行输入五个整数A，B，k1，k2，mod。

第二行输入一个整数Q。

接下来Q行每行输入一个整数pos。

1<=A<=100
1<=B<=100
2<=k1,k2<=1e14
1<=mod<=1e9+7
1<=Q<=100000
1<=pos<=1e14

输出

输出Q行，每行表示a[pos]%mod。

样例输入

45 87 2 5 1000000007
11
0
1
2
3
4
5
6
7
8
9
10

样例输出

45
87
3915
3915
176175
340605
340605
340605
15327225
15327225
333468568

#include <bits/stdc++.h>
using namespace std;

int A, B;
long long k1, k2;
int mod;

map<long long, int>mp;

long long F(long long i)
{
	if (mp.count(i) == 1)
		return mp[i];

	if (i == 0)
		return mp[i] = A;
	else if (i == 1)
		return mp[i] = B;

	return mp[i] = (F(i / k1) * F(i / k2)) % mod;
}

int main()
{
	scanf("%d%d%lld%lld%d", &A, &B, &k1, &k2, &mod);
	
	int Q;
	scanf("%d", &Q);
	while (Q--)
	{
		long long pos;
		scanf("%lld", &pos);

		F(pos);

		printf("%d\n", mp[pos]);
	}

	return 0;
}