题意分析:
这道题的题意很容易让人理解,可能一些人会想如果这道题的R,G的取值范围是100,那么一个递归调用就可以解出来了(这你妹的,非要来了200005)。后来看了别人的解题报告之后,才明白了。这道题的解题思路可以是这样:先算出来有堆叠多少层,设为H,现假设这里有J个红木,已堆叠到了I层,那么DP[I][J]就表示堆叠到I层时有J个红木时可以有多少种堆法,所以DP[I][J]+=DP[I-1][J-I];因为这是一个叠加过程,所以可以用滚动数组。
代码如下:
#include<iostream>
#include<cstdio>
#define mo 1000000007
using namespace std;
long long dp[200005],ans=0;
int main()
{
long long r,g,i;
cin>>r>>g;
long long h;
for(h=1; h*(h+1)/2<=(r+g); h++);
h--;
dp[0] = 1;
long long mi=h*(h+1)/2,ans=0;
for(i=1; i<=h; i++) {
for(int j=r; j>=0; j--) {
if(j-i>=0) dp[j]=(dp[j]+dp[j-i])%mo;
}
}
for(i=0; i<=r; i++) {
if(mi-i<=g) ans=(ans+dp[i])%mo;
}
cout<<ans<<endl;
return 0;
}