题解:
设
x=∑mi=1k−ai,y=∑ni=1k−bi
x
=
∑
i
=
1
m
k
−
a
i
,
y
=
∑
i
=
1
n
k
−
b
i
,合法解
x
x
满足。
也就是如果 x x 能被表示为,且 1−x 1 − x 能被表示为 ∑ni=1k−bi ∑ i = 1 n k − b i ,这个 x x 就对应一种方案。
直接在进制意义下做DP即可,设最后一位位于 −t − t 位,各位的总和为 val v a l ,那么要满足 val≤m,val≡m(modk−1) v a l ≤ m , v a l ≡ m ( mod k − 1 ) ,且 t(k−1)−val+1≤n,t(k−1)−val+1≡n(modk−1) t ( k − 1 ) − v a l + 1 ≤ n , t ( k − 1 ) − v a l + 1 ≡ n ( mod k − 1 ) 。 按位DP即可。
#include <bits/stdc++.h>
using namespace std;
const int N=4e3+50, mod=1e9+7;
inline int add(int x,int y) {return (x+y>=mod) ? (x+y-mod) : (x+y);}
inline int dec(int x,int y) {return (x-y<0) ? (x-y+mod) : (x-y);}
int n,m,k,ans,s[N],f[N];
int main() {
cin>>n>>m>>k;
for(int i=0;i<=m;i++) s[i]=1;
for(int i=1;i<=(n+m)/(k-1);++i) {
for(int j=1;j<=m;++j) {
f[j]=dec(s[j-1],(j>=k) ? s[j-k] : 0);
if((j%(k-1))==(m%(k-1)) && 1+i*(k-1)-j<=n && ((1+i*(k-1)-j)%(k-1))==(n%(k-1))) ans=add(ans,f[j]);
}
for(int j=1;j<=m;j++) f[j]=add(f[j-1],f[j]), s[j]=add(s[j],f[j]);
} cout<<ans<<'\n';
}