P1630 求和（快速幂&规律）

求和

题目描述

求 $1^b+2^b+\cdots +a^b$ 的和除以 $10^4$ 的余数。

输入格式

本题有多组数据。

第一行一个整数 $N$，表示共有 $N$ 组测试数据。

对于每组数据，一行两个整数 $a,b$。

输出格式

对于每组数据，一行一个整数，表示答案。

样例 #1

样例输入 #1

1
2 3

样例输出 #1

9

提示

对于 $30\%$ 的数据，$N\le 10$，$a,b\le 10^3$。

对于 $100\%$ 的数据，$1\le N\le 100$，$1\le a,b\le 10^9$。

解析

可以观察到，将a分为多段，其中包含多段1-10000和一段10000以内的段。
前缀和，快速幂求出10000以内的前缀和，然后计算即可

#include<bits/stdc++.h>
using namespace std;
#define int long long
const int mod=1e4;
const int N=1e4+5;
int a,b,sum[N];
int qmi(int m,int k,int p){		
	int res=1%p,t=m;
	while(k){
		if(k&1) res=res*t%p;
		t=t*t%p;
		k>>=1;
	}
	return res%mod;
}
void solve(){
	scanf("%lld%lld",&a,&b);
	for(int i=1;i<=1e4;i++){
		sum[i]=sum[i-1]+qmi(i,b,mod);
		sum[i]%=mod;
	}
	printf("%lld\n",(a/10000*sum[10000]+sum[a%10000])%mod);
}
signed main(){
	int t=1;
	scanf("%lld",&t);
	while(t--) solve();
	return 0;
}