4591: [Shoi2015]超能粒子炮·改
题意:多组询问,求
[S(n, k) = sum_{i=0}^n inom{n}{i} mod 2333, k le n le 10^{18}
]
lucas定理,展开一层然后整除分块一下,不完整的块单独拿出来,就是
[S(n,k) = S(frac{n}{p}, frac{k}{p}-1)S(n mod p, p-1) + inom{frac{n}{p}}{ frac{k}{p}} S(n mod p, k mod p)
]
预处理(n,k le 2333)的
单次询问复杂度log^2
#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <cmath>
using namespace std;
typedef long long ll;
const int N = 2500, P = 2333;
inline ll read() {
char c=getchar(); ll x=0,f=1;
while(c<'0'||c>'9') {if(c=='-')f=-1; c=getchar();}
while(c>='0'&&c<='9') {x=x*10+c-'0'; c=getchar();}
return x*f;
}
ll n, k;
int c[N][N], s[N][N];
inline ll lucas(ll n, ll m) {
if(n < m) return 0;
ll ans = 1;
for(; m; n /= P, m /= P) ans = ans * c[n % P][m % P] %P;
return ans;
}
ll S(ll n, ll k) {
if(n <= P && k <= P) return s[n][k];
ll ans = (S(n / P, k / P - 1) * S(n % P, P - 1) + lucas(n / P, k / P) * S(n % P, k % P)) %P;
return ans;
}
int main() {
freopen("in", "r", stdin);
int T = read();
c[0][0] = 1;
for(int i=1; i<=P; i++) {
c[i][0] = 1;
for(int j=1; j<=i; j++) c[i][j] = (c[i-1][j] + c[i-1][j-1]) %P;
}
for(int i=0; i<=P; i++) {
s[i][0] = c[i][0];
for(int j=1; j<=P; j++) s[i][j] = (s[i][j-1] + c[i][j]) %P;
}
while(T--) {
n = read(); k = read();
printf("%lld
", S(n, k));
}
}