Codeforces 478D Red-Green Towers 构造+DP

题意:用r个红方块和g个绿方块构造高度为h的楼时,要满足第i层有i个方块,并且相同层的方块颜色相同.方块可以不全部用完
r,g<=2e5 求h的最大值对应的方法数.

首先最大高度h(1+h)<=2*(r+g) 可以简单的构造出最大高度h.高度h可以构造出1~h(h+1)/2的任意一个数.
所以当红色用r个时,剩下sum-r肯定小于等于g 把g按照若干组分割,得到一个合法法案.
已知高度h,剩下就考虑那些层涂红色,若红色用了x个 则要知道[1,h]个数中 选某些个数其和为x的方法数

并且剩下sum-x个要<=g. dp[i][x]=d[i-1][x]+d[i-1][x-i]

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const ll mod=1e9+7;
const int N=2e5+5;
ll a,b,d[N];
int main()
{
	while(cin>>a>>b)
	{
		if(a>b)
			swap(a,b);
		ll h=1,x=1;
		while(a+b>=x)	
			h++,x=(1+h)*h/2;
		h--;
		ll sum=(1+h)*h/2;
		//cout<<h<<endl, hmx=893;
		memset(d,0,sizeof(d));
		d[0]=1;
		for(int i=1;i<=h;i++)
		{
			for(int j=a;j>=i;j--)
			{
				d[j]=(d[j]+d[j-i])%mod;
			}
		}
		ll ans=0;
		for(int i=0;i<=a;i++)
		{
			if(sum-i<=b)
				ans=(ans+d[i])%mod;
		} 
		printf("%I64d\n",ans);
	}
	return 0;
}