正题
题目链接:https://www.luogu.com.cn/problem/P4345
题目大意
\(T\)组询问，给出\(n,k\)求
\[\sum_{i=0}^{k}\binom{n}{i}
\]
对\(2333\)取模的值
\(1\leq T\leq 10^5,1\leq k\leq n\leq 10^{18}\)
解题思路
因为模数很小，可以考虑用\(Lucas\)定理，然后考虑怎么优化复杂度。
对于给出的\(n,k\)分成两个部分，第一部分是由\(k\)前面若干段长度为\(P\)的整段构成，这一部分的答案我们发现对于\(C_{\lfloor\frac{n}{P}\rfloor}^{\lfloor\frac{m}{P}\rfloor}\times C^{n\%p}_{m\% p}\)这两个值，后面那一个值的和是确定的，是\(\sum_{i=1}^kC_{n\%p}^k\)，前面那一部分的值我们可以递归下去计算。
然后第二部分是剩下的散段，这个部分我们也是自直接递归下去算就可以了
时间复杂度\(O(T\log n)\)
code
#include<cstdio>
#include<cstring>
#include<algorithm>
#define ll long long
using namespace std;
const ll P=2333;
ll n,k,t,S[P][P],C[P][P];
ll Lucas(ll n,ll k){
if(!k)return 1ll;
if(!C[n%P][k%P])return 0;
return Lucas(n/P,k/P)*C[n%P][k%P]%P;
}
ll solve(ll n,ll k){
if(k<0)return 0;
if(n<P)return S[n][min(n,k)];
ll tmp=solve(n/P,k/P-1)*S[n%P][n%P]%P;
tmp=(tmp+solve(n%P,k%P)*Lucas(n/P,k/P)%P)%P;
return tmp;
}
signed main()
{
C[0][0]=S[0][0]=1;
for(ll i=1;i<P;i++)
for(ll j=0;j<=i;j++)
C[i][j]=((j?C[i-1][j-1]:0)+C[i-1][j])%P;
for(ll i=1;i<P;i++){
S[i][0]=C[i][0];
for(ll j=1;j<=i;j++)
(S[i][j]=S[i][j-1]+C[i][j])%=P;
}
scanf("%lld",&t);
while(t--){
scanf("%lld%lld",&n,&k);
printf("%lld\n",solve(n,k));
}
return 0;
}