Codeforces478D-Red-Green Towers-DP

不是特别难的一道dp题。

给r个红块，g个绿块，计算这些块能磊出的最高塔的方案数。

塔的每一层都比上一层多一块，每一层只能有一种颜色。

dp[i][j]表示第i层，j个红块的方案数。

则dp[i][j] = dp[i-1][j] + dp[i-1][j-i].注意一下方案的转移和最终结果的统计。

 #include <cstdio>
 #include <cstring>
 #include <algorithm>

 using namespace std;

 const int maxn = 2e5+;
 const int mod = 1e9+;

 int r,g;
 int dp[maxn];

 int main()
 {
     scanf("%d%d",&r,&g);
     memset(dp,,sizeof dp);

     int ans = ,oans;
     if(r > ) dp[]++;
     if(g > ) dp[]++;
     oans = dp[]+dp[];
     for(int i=;i<;i++)
     {
         bool flag = false;
         int tol = i*(i+)/;
         int bgn = min(r,tol);
         ans = ;

         for(int j=bgn;j>=;j--)
         {
             bool flag_g = false;
             if((tol-j) <= g && dp[j])
             {
                 dp[j] = dp[j];
                 dp[j] %= mod;
                 flag = true;
                 flag_g = true;
                 ans += dp[j];
                 ans %= mod;
             }
             if(j >= i && dp[j-i])
             {
                 dp[j] += dp[j-i];
                 dp[j] %= mod;
                 flag = true;
                 flag_g = true;
                 ans += dp[j-i];
                 ans %= mod;
             }
             if(!flag_g)
                 dp[j] = ;
             //printf("i:%d j:%d dp:%d\n",i,j,dp[j]);
         }
         if(!flag)
         {
             break ;
         }
         oans = ans;
     }
     printf("%d\n",oans);
 }