题目描述
组合数Cnm表示的是从n个物品中选出m个物品的方案数。举个例子,从(1,2,3) 三个物品中选择两个物品可以有(1,2),(1,3),(2,3)这三种选择方法。根据组合数的定 义,我们可以给出计算组合数的一般公式:
Cnm=m!(n−m)!n!
其中n! = 1 × 2 × · · · × n
小葱想知道如果给定n,m和k,对于所有的0 <= i <= n,0 <= j <= min(i,m)有多少对 (i,j)满足Cij是k的倍数。
输入输出格式
输入格式:
第一行有两个整数t,k,其中t代表该测试点总共有多少组测试数据,k的意义见 【问题描述】。
接下来t行每行两个整数n,m,其中n,m的意义见【问题描述】。
输出格式:
t行,每行一个整数代表答案。
输入输出样例
输入样例#1:
1 2 3 3
输出样例#1:
1
输入样例#2:
2 5 4 5 6 7
输出样例#2:
0 7
说明
【样例1说明】
在所有可能的情况中,只有C21=2是2的倍数。
【子任务】
TLE了两个点
没想好前缀和怎么写【叹气
90分~~
#include<bits/stdc++.h>
using namespace std;
template <typename T> void read(T &x){
x=0;int f=1;char ch=getchar();
for(;!isdigit(ch);ch=getchar())if(ch=='-')f=-1;
for(;isdigit(ch);ch=getchar())x=(x<<1)+(x<<3)+ch-'0';
x*=f;
}
typedef long long LL;
int T,K;
int n,m;
LL f[2010][2010];
int main(){
read(T),read(K);
memset(f,0,sizeof(f));
f[1][1]=1;
for(int i=1;i<=2001;++i){
f[i][1]=i%K;
f[i][i]=1;
}
for(int i=2;i<=2001;++i)
for(int j=2;j<=i-1;++j){
f[i][j]=f[i-1][j-1]+f[i-1][j];
if(f[i][j]>=K) f[i][j]%=K;
}
/*for(int i=1;i<=6;++i){
for(int j=1;j<=7;++j)cout<<"dp["<<i<<"]["<<j<<"]="<<dp[i][j]<<" ";
cout<<endl;
}*/
for(int t=1;t<=T;++t){
read(n),read(m);
LL ans=0;
/*for(int i=1;i<=n;++i){
for(int j=1;j<=m;++j) cout<<"f["<<i<<"]["<<j<<"]="<<f[i][j]<<" ";
cout<<endl;
}*/
for(int i=1;i<=n;++i){
for(int j=1;j<=min(i,m);++j)
if(f[i][j]==0){
//cout<<i<<" "<<j<<endl;
ans++;
}
}
cout<<ans<<endl;
}
return 0;
}
加了一个比较弱的前缀和 95分
#include<bits/stdc++.h>
using namespace std;
template <typename T> void read(T &x){
x=0;int f=1;char ch=getchar();
for(;!isdigit(ch);ch=getchar())if(ch=='-')f=-1;
for(;isdigit(ch);ch=getchar())x=(x<<1)+(x<<3)+ch-'0';
x*=f;
}
typedef long long LL;
int T,K;
int n,m;
LL f[2010][2010];
LL dp[2010][2010];
int main(){
read(T),read(K);
memset(f,0,sizeof(f));
memset(dp,0,sizeof(dp));
f[1][1]=1;
for(int i=1;i<=2001;++i){
f[i][1]=i%K;
f[i][i]=1;
}
for(int i=2;i<=2001;++i)
for(int j=2;j<=i-1;++j){
f[i][j]=f[i-1][j-1]+f[i-1][j];
if(f[i][j]>=K) f[i][j]%=K;
}
for(int i=1;i<=2001;++i){
if(f[i][1]==0) dp[i][1]=1;
for(int j=2;j<=i;++j)
if(f[i][j]==0) dp[i][j]=dp[i][j-1]+1;
else dp[i][j]=dp[i][j-1];
for(int k=i+1;k<=2001;++k)
dp[i][k]=dp[i][k-1];
}
/*for(int i=1;i<=6;++i){
for(int j=1;j<=7;++j)cout<<"dp["<<i<<"]["<<j<<"]="<<dp[i][j]<<" ";
cout<<endl;
}*/
for(int t=1;t<=T;++t){
read(n),read(m);
LL ans=0;
/*for(int i=1;i<=n;++i){
for(int j=1;j<=m;++j) cout<<"f["<<i<<"]["<<j<<"]="<<f[i][j]<<" ";
cout<<endl;
}*/
/*for(int i=1;i<=n;++i){
for(int j=1;j<=min(i,m);++j)
if(f[i][j]==0){
//cout<<i<<" "<<j<<endl;
ans++;
}
}
cout<<ans<<endl;*/
for(int i=1;i<=n;++i) ans+=dp[i][m];
cout<<ans<<endl;
}
return 0;
}
二维前缀和(标算
#include<bits/stdc++.h>
using namespace std;
template <typename T> void read(T &x){
x=0;int f=1;char ch=getchar();
for(;!isdigit(ch);ch=getchar())if(ch=='-')f=-1;
for(;isdigit(ch);ch=getchar())x=(x<<1)+(x<<3)+ch-'0';
x*=f;
}
typedef long long LL;
int T,K;
int n,m;
LL f[2010][2010];
LL dp[2010][2010];
int main(){
read(T),read(K);
memset(f,0,sizeof(f));
memset(dp,0,sizeof(dp));
f[1][1]=1;
for(int i=1;i<=2001;++i){
f[i][1]=i%K;
f[i][i]=1;
}
for(int i=2;i<=2001;++i)
for(int j=2;j<=i-1;++j){
f[i][j]=f[i-1][j-1]+f[i-1][j];
if(f[i][j]>=K) f[i][j]%=K;
}
/*for(int i=1;i<=2001;++i){
if(f[i][1]==0) dp[i][1]=1;
for(int j=2;j<=i;++j)
if(f[i][j]==0) dp[i][j]=dp[i][j-1]+1;
else dp[i][j]=dp[i][j-1];
for(int k=i+1;k<=2001;++k)
dp[i][k]=dp[i][k-1];
}*/
for(int i=2;i<=2001;++i){
for(int j=1;j<=i;++j){
if(f[i][j]==0) dp[i][j]=dp[i-1][j]+dp[i][j-1]-dp[i-1][j-1]+1;
else dp[i][j]=dp[i-1][j]+dp[i][j-1]-dp[i-1][j-1];
}
for(int k=i+1;k<=2001;++k)
dp[i][k]=dp[i][k-1];
}
/*for(int i=1;i<=6;++i){
for(int j=1;j<=7;++j)cout<<"dp["<<i<<"]["<<j<<"]="<<dp[i][j]<<" ";
cout<<endl;
}*/
for(int t=1;t<=T;++t){
read(n),read(m);
LL ans=0;
/*for(int i=1;i<=n;++i){
for(int j=1;j<=m;++j) cout<<"f["<<i<<"]["<<j<<"]="<<f[i][j]<<" ";
cout<<endl;
}*/
/*for(int i=1;i<=n;++i){
for(int j=1;j<=min(i,m);++j)
if(f[i][j]==0){
//cout<<i<<" "<<j<<endl;
ans++;
}
}
cout<<ans<<endl;*/
/*for(int i=1;i<=n;++i) ans+=dp[i][m];
cout<<ans<<endl;*/
cout<<dp[n][m]<<endl;
}
return 0;
}