Codeforces Round #273 (Div. 2) D. Red-Green Towers

果断DP

第一想法是二维dp，第一维表示有多少个红色，第二维表示有多少个绿色，但是空间消耗太大。因此考虑表示红色的层数，绿色的层数，但是你没法根据层数得到你剩余的数目。因此必须记录某一个颜色的数目，又要记录层数。从而产生了dp[ i ][ j ] 公式，i 表示层数，j 表示红色（绿色也行）的数目。

可以根据层数、现在红色的个数得到，已经用了多少绿色，再分别判断红色和绿色是否可以增加一层（个数大于 i+1 ）;

再判断这一层是否转移到了，转移到了就可以跟新 ans ，没转移到就break

//      whn6325689
#include <algorithm>
#include <iostream>
#include <iomanip>
#include <cstring>
#include <climits>
#include <complex>
#include <fstream>
#include <cassert>
#include <cstdio>
#include <bitset>
#include <vector>
#include <deque>
#include <queue>
#include <stack>
#include <ctime>
#include <set>
#include <map>
#include <cmath>


using namespace std;

typedef long long ll;
typedef long double ld;
typedef pair<ll, ll> pll;
typedef complex<ld> point;
typedef pair<int, int> pii;
typedef pair<pii, int> piii;
typedef vector<int> vi;

#define CLR(x,y) memset(x,y,sizeof(x))
#define mp(x,y) make_pair(x,y)
#define pb(x) push_back(x)
#define lowbit(x) (x&(-x))
#define MID(x,y) (x+((y-x)>>1))
#define eps 1e-9
#define INF 0x3f3f3f3f
#define LLINF 1LL<<62

template<class T>
inline bool read(T &n)
{
    T x = 0, tmp = 1; char c = getchar();
    while((c < '0' || c > '9') && c != '-' && c != EOF) c = getchar();
    if(c == EOF) return false;
    if(c == '-') c = getchar(), tmp = -1;
    while(c >= '0' && c <= '9') x *= 10, x += (c - '0'),c = getchar();
    n = x*tmp;
    return true;
}
template <class T>
inline void write(T n)
{
    if(n < 0)
    {
        putchar('-');
        n = -n;
    }
    int len = 0,data[20];
    while(n)
    {
        data[len++] = n%10;
        n /= 10;
    }
    if(!len) data[len++] = 0;
    while(len--) putchar(data[len]+48);
}
//-----------------------------------]

const int MOD=1000000007;
const int MAXN=1000;
int num[MAXN];
int dp[2][200005],r,g;

int main()
{
	for(int i=1;i<MAXN;i++)
		num[i]=num[i-1]+i;
	read(r),read(g);
	dp[0][0]=1;
    int now=1;
    int ans=0;
	for(int i=0;i<MAXN;i++,now^=1)
	{
		for(int j=0;j<=r;j++)
			dp[now][j]=0;
		for(int j=0;j<=r;j++)
			if(dp[now^1][j])
			{
				int ways=dp[now^1][j];
                int left_r=r-j;
                int left_g=g-(num[i]-j);
				if(left_r>=i+1)
					dp[now][j+i+1]=(dp[now][j+i+1]+dp[now^1][j])%MOD;
				if(left_g>=i+1)
					dp[now][j]=(dp[now][j]+dp[now^1][j])%MOD;
			}
		int sum=0,ok=0;
		for(int j=0;j<=r;j++)
            if(dp[now][j])
            {
                ok=1;
                sum=(sum+dp[now][j])%MOD;
            }
        if(ok)
            ans=sum;
        else
            break;
	}
	write(ans),putchar('\n');
	return 0;
}

还有一份更加简洁的思路

dp[ i ] 表示在某一层用了多少红色

在最高层的时候break，那么dp就恰好表示最高层需要的数目

再判断绿色是否符合，如果符合就加到答案上

//      whn6325689
#include <algorithm>
#include <iostream>
#include <iomanip>
#include <cstring>
#include <climits>
#include <complex>
#include <fstream>
#include <cassert>
#include <cstdio>
#include <bitset>
#include <vector>
#include <deque>
#include <queue>
#include <stack>
#include <ctime>
#include <set>
#include <map>
#include <cmath>


using namespace std;

typedef long long ll;
typedef long double ld;
typedef pair<ll, ll> pll;
typedef complex<ld> point;
typedef pair<int, int> pii;
typedef pair<pii, int> piii;
typedef vector<int> vi;

#define CLR(x,y) memset(x,y,sizeof(x))
#define mp(x,y) make_pair(x,y)
#define pb(x) push_back(x)
#define lowbit(x) (x&(-x))
#define MID(x,y) (x+((y-x)>>1))
#define eps 1e-9
#define INF 0x3f3f3f3f
#define LLINF 1LL<<62

template<class T>
inline bool read(T &n)
{
    T x = 0, tmp = 1; char c = getchar();
    while((c < '0' || c > '9') && c != '-' && c != EOF) c = getchar();
    if(c == EOF) return false;
    if(c == '-') c = getchar(), tmp = -1;
    while(c >= '0' && c <= '9') x *= 10, x += (c - '0'),c = getchar();
    n = x*tmp;
    return true;
}
template <class T>
inline void write(T n)
{
    if(n < 0)
    {
        putchar('-');
        n = -n;
    }
    int len = 0,data[20];
    while(n)
    {
        data[len++] = n%10;
        n /= 10;
    }
    if(!len) data[len++] = 0;
    while(len--) putchar(data[len]+48);
}
//-----------------------------------

const int N=200007;
const int MOD=1000000007;

int dp[N]={1};
int main()
{
    int r,g,ans=0,need=0;
    read(r),read(g);
    for(int i=1;;i++)
    {
        need+=i;
        if(need>r+g)
        {
            need-=i;
            break;
        }
        for(int j=min(r,need);j>=i;j--)
            dp[j]=(dp[j]+dp[j-i])%MOD;
    }
    for(int x=max(0,need-g);x<=min(need,r);x++)
        ans=(ans+dp[x])%MOD;
    write(ans),putchar('\n');
    return 0;
}