5875: 桃子的难题

5875: 桃子的难题 

时间限制(普通/Java):1000MS/3000MS     内存限制:65536KByte
 

描述

taozi喜欢数学，但是遇到数学题就头疼，zdragon为了让大家高兴高兴，给taozi出了道难题：

S=∑q^i (1≤i≤n)，由于答案可能会很大，答案对p取模。

输入

输入第一行为测试样例组数T（1≤T≤100）。

对于每组数据第一行包含三个正整数n，q，p(1≤n，q，p≤109)。

输出

对于每组数据，输出一个S对p取模的值。

样例输入

2
3 2 100
4 511 520

样例输出

14
184

提示

对于第一个样例，21+22+23=14，对100取模，答案为14。

 

S=∑q^i (1≤i≤n)

首先看到这个公式，若果直接用快速幂的话肯定超时

然后我们可以想到等比数列求和

当你准备使用这个公式的时候你会发现有一个问题，就是取模的话怎么办，取模除法是不满足拆分条件的

如

那么这里肯定是没有办法的

所以我们可以根据如下思路：

就是1+q^1+q^2+....q^n

然后当n是偶数的时候：

把这个式子分成前后相等长度的两部分

（p^0+p^1+..p^(n/2)）+ (p^(n/2+1)+p(n/2+2)+...+p^n)

把后半部分的p^(n/2+1)提取出来。

（p^0+p^1+..p^(n/2)) * (p^(n/2+1)+1)

右边快速幂，左边递归

奇数的话去掉p^n，同样的操作去左边递归右边快速幂。

如：

n为奇=3，q1+q2+q3=(q1)+q2+q2(q1)=(q1)*(1+q2)+q2

n为偶=4，q1+q2+q3+q4=(q1+q2)+q2(q1+q2)=(q1+q2)*(1+q2)

可以发现无论奇偶都是提出一个数然后只用算前半段。

参考：https://www.cnblogs.com/Esquecer/p/10993981.html

代码:

#include<iostream>
#include<cstring>
using namespace std;
typedef long long ll;
const int MAXN = 1e9+10;

int T;
ll N,P,Q;

ll get_ans(ll num,ll Q,ll P){
	
	ll temp_num = num;
	ll ans = 1;
	ll C = Q%P;
	while(temp_num>0){
		if(temp_num&1) ans= ans*C%P;
		C=C*C%P;
		temp_num = temp_num>>1;
	}
	return ans%P;
} 

ll get_sum(ll num, ll Q, ll P){
	if(num == 1 ) return Q%P;
	if(num == 0 ) return 1;
	
	int temp = num;
	ll ans = 0;
	if(num&1){
		temp--;
		ans = get_sum(temp/2,Q,P)*(get_ans(temp/2,Q,P)+1)%P + get_ans(num,Q,P)%P;
	}else{
		ans = get_sum(temp/2,Q,P)*(get_ans(temp/2,Q,P)+1)%P;
	}

	return ans;
}

int main(){
	
	scanf("%d",&T);
	while(T--){
		scanf("%lld%lld%lld",&N,&Q,&P);

		ll ans = get_sum(N,Q,P);
		ans = (ans%P+P)%P;
		printf("%lld\n",ans);
	}
	
	return 0;
}

主要核心在于分治思想