Submission #2211438

---

Source Code Expand

#include <iostream>
#include <cstdio>
#include <cstring>
using namespace std;
const int mod=1e9+7;
int n,m,K,f[4005][2005][2],sum[4005];
int add(int a,int b){return a+b>=mod?a+b-mod:a+b;}
void upd(int &a,int b){a=(a+b>=mod)?a+b-mod:a+b;}
int main(){
	scanf("%d%d%d",&n,&m,&K);
	int tot=(n+m-1)/(K-1);
	f[0][0][0]=1;
	n--;K--;
	int ans=0;
	for(int i=1;i<=tot;i++){
		for(int p=1;p<=n+1;p++) sum[p]=add(sum[p-1],add(f[i-1][p-1][0],f[i-1][p-1][1]));
		for(int p=0;p<=n;p++){
			f[i][p][0]=add(sum[p+1],mod-sum[p]);
			int k=max(0,p-K);
			f[i][p][1]=add(sum[p],mod-sum[k]);
		}
		for(int p=n%K;p<=n;p+=K){
			if(i*K-p<=m) upd(ans,f[i][p][1]);
		}
	}
	printf("%d",ans);
	return 0;
}

Submission Info

Compile Error

./Main.cpp: In function ‘int main()’:
./Main.cpp:10:26: warning: ignoring return value of ‘int scanf(const char*, ...)’, declared with attribute warn_unused_result [-Wunused-result]
  scanf("%d%d%d",&n,&m,&K);
                          ^

Judge Result