题意:给定两种颜色的red,blue的个数方块,要求你把他们拼成如上图的图形(第一层为1个,第二层二2个,。。。第n层为n个),而且每层的颜色必须相同,你可以不适用完方块,满足图形在最高层时,
最多能拼的方案数,如上图红色为4个蓝色为6个,方案数为2。
刚开始拿到题时感觉想歪了,例如上例我首先想到4或者6有多少个(1---4(4为最大层数))里可以拼起来,一种方案中每个数只能算一次例如6(4+2,1+2+3)答案为2 ,但是不知怎么下笔。其实如果我们把层数当做dp的一维,需要分解的数当做一维,解决方案慢慢就浮出来了。。。。。,之前就是忽略的层数。。。。。。。,
以后做类似的题需要往多个角度想想,不要局限一个地方。
其实可以知道某一个颜色可以放的个数范围从min(n)=【h*(h+1)/2 - m】, max(n)=MAX(n,h*(h+1)/2);
然后dp计算该区间的值,总和即为答案。
dp:d[i][j]=d[i-1][j]+d[i-1][j-i]; i表示前i层,放j个red的方案数(因为前i层方块总数已确定,确定了red的个数,那剩下来来的就是blue),他有两种情况组成,当第i层放red时,
就是前i-1层放j-i个red方块(已经用掉i个方块);,当第i层放blue时,就是前i-1层放j个red方块;
#include<stdio.h>
#include<string.h>
#include<math.h>
#include<iostream>
#include<algorithm>
using namespace std;
const int mod=1e9+7;
long long d[400010];
int main()
{
int n,m,i,j,s,k;
long long ans;
while(scanf("%d%d",&n,&m)!=EOF)
{
s=m+n;
ans=0;
k = int(sqrt(s*2.0))+1;
while(k*(k+1)>2*s)
k--;
//printf("k=%d\n",k);
// for(i=0;i<200010;i++)
// d[i]=0;
memset(d,0,sizeof(d));
d[0]=1;d[1]=1;
for(i=2;i<=k;i++)
{
for(j=min(i*(i+1)/2,n);j>=0;j--)
{
if(j>=i)
d[j]=(0LL+d[j]+d[j-i])%mod;
}
//d[i][j]=(1LL+d[i-1][j]+d[i][j-i])%mod;
}
for(i=max(k*(k+1)/2-m,0);i<=n;i++)
{
//printf("%d ",d[i]);
ans=(0LL+ans+d[i])%mod;
}
printf("%I64d\n",ans);
}
}