2022年ICPC网络赛第二场 A(找规律加费马小定理)

题目
题意: 给定一个n位的数(至多10^6)，但是不是直接给出。而是给定min(100,n)行，第i行有i个数，第j个数表示[j,j+i,j+2i…]数之和，如果超出了n那就不算了。给定一个质数p (p是小于等于97的所有质数，并且p不等于5).
思路: 当时单挑的，也没脑子想了。其实算一算，1001e610，就算把数求出来也没用。多想想为什么p<=97且p不等于5.
  注意到p不等于5，想到费马小定理，x^(p-1) === 1(mod p). 太对了，如果p=5，就没法找循环了。因为10 ^(p-1)在模p意义下就没了。这时我们发现是有循环节的，每p个数，作为一循环。那么对于位置为idx的数，与他10的幂次相同的有idx+p,idx+2p…直接用题目给定第p-1行数据就能搞出来。
  另外，有可能p-1是大于n的，而n<=100时数直接给出了，直接求就行，记得特判一下。
感觉这题出的贼好
代码:

#include<bits/stdc++.h>
using namespace std;
const int N = 1e6+10;
#define int long long
int a[102][102];
int n,m,k,T;
int qpow(int a,int k,int p)
{
	int res = 1;
	while(k)
	{
		if(k&1) res = res * a % p;
		k >>= 1;
		a = a * a % p;
	}
	return res;
}
void solve()
{
	cin>>n;
	for(int i=1;i<=min(1ll*100,n);++i)
	{
		for(int j=1;j<=i;++j)
		{
			cin>>a[i][j];
		}
	}
	
	cin>>k;
	while(k--)
	{
		int res = 0;
		int p; cin>>p;
		if(n<=100)
		{
			for(int j=1;j<=n;++j)
			{
				int t = qpow(10,n-j,p);
				t = t * a[n][j] % p;
				res += t; res %= p;
			}
		}
		else
		{
			int wh = p-1; //循环节
			for(int i=1;i<=wh;++i)
			{
					int ans = qpow(10,(n-i)%wh,p);
					int sum = a[wh][i] % p;
					sum = sum * ans % p;
					res += sum; res %= p;
			}
		}
		res %= p;
		res = (res + p) % p;
		cout<<res<<"\n";
	}
}
signed main(void)
{
	cin>>T;
	while(T--)
	solve();
	return 0;
}
/*
1
9
47
27 20
14 18 15
16 13 11 7
13 12 13 5 4
12 14 11 2 4 4
14 12 8 2 4 4 3
12 9 8 2 4 4 3 5
9 9 8 2 4 4 3 5 3
2
3
7
*/