D. Red-Green Towers（dp+滚动数组)

https://codeforces.com/problemset/problem/478/D

---

思路：

暴力的话n,i,j三层。直接炸了阿..

但是层数的话不会太多，估摸着最多1000层，i，j的话其实存一个就好。因为当前总数能算，i有一个就能算j。

dp[i][j]表示从上到下摆到第i层用j个红格子的方法数

第i层摆红格子:dp[i][j]+=dp[i-1][j-i];
摆绿格子的话，已经摆的绿格子为i*（i-1）/2 -j  ,看剩下的能不能摆。能摆的话dp[i][j]+=dp[i-1][j]。

但是这样空间还是不够的。而且这题挺卡常的如果优化不当。

因为开始我是直接滚动两维优化tle11.

 LL pre=(i+1)*i/2;///优化
      for(int j=0;j<=r&&j<=pre;j++){

加了这句跑到1100ms。

但是不用滚动的两维直接变一维只要200ms.因为他不用常数的每次清空。效率很高。

#include<iostream>
#include<vector>
#include<queue>
#include<cstring>
#include<cmath>
#include<map>
#include<set>
#include<cstdio>
#include<algorithm>
#define debug(a) cout<<#a<<"="<<a<<endl;
using namespace std;
const int maxn=2e5+1000;
typedef long long LL;
const LL mod=1e9+7;
inline LL read(){LL x=0,f=1;char ch=getchar();	while (!isdigit(ch)){if (ch=='-') f=-1;ch=getchar();}while (isdigit(ch)){x=x*10+ch-48;ch=getchar();}
return x*f;}
LL dp[2][maxn];
int main(void){
   LL r,g;r=read();g=read();
   int n=1000;
   while(n*(n+1)/2>(r+g)){
        n--;
   }
   dp[0][0]=1;
   for(int i=1;i<=n;i++){///从上往下第一层开始,当前第i层
      memset(dp[i&1],0,sizeof(dp[i&1]));///注意当前层清空更新这层答案
      LL pre=(i+1)*i/2;///优化
      for(int j=0;j<=r&&j<=pre;j++){
          ///该层用红色
          if(j>=i) dp[i&1][j]=(dp[i&1][j]+dp[(i-1)&1][j-i])%mod;
          ///该层用绿色
          LL num=i*(i-1)/2-j;///已经摆的绿格子
          if(g-num>=i){
             dp[i&1][j]=(dp[i&1][j]+dp[(i-1)&1][j])%mod;
          }
      }
   }
   LL ans=0;
   for(int i=0;i<=r;i++){
       ans=(ans+dp[n&1][i])%mod;
   }
   printf("%lld\n",ans);
   return 0;
}

 

---

直接一维的参考：https://blog.csdn.net/ffgcc/article/details/81056429

 

    dp[0]=1;
    for(int i=1;i<=h;i++)
    {
        for(int j=r;j>=i;j--)//逆序的滚动数组
        {
            dp[j]=(dp[j-i]+dp[j])%mod;
        }
    }
    int mx=0;
    for(int j=0;j<=r;j++)
    {
        if(sum-j<=g)
            mx+=dp[j],mx%=mod;
    }