vijos 1417 魔法塔防 - DP

传送门

题解：显然红塔放在最后。

dp[i][j]表示前i座塔放了几个绿塔，剩下的放红塔。

这样是O(n^2)的。

此题显然可以做到O(gb+n)的复杂度。

可以通过形如n<=1e5,g,b<=1e3,r<=1e5的数据。

发现可以优化空间。

//vijos 1411
#include<iostream>
#include<cstring>
#include<cstdio>
#include<algorithm>
#define lint long long
#define N 1100
using namespace std;
lint dp[2][N];int R,G,B,t;
inline lint calc(int r,int g,int b,int R=0)
{
	lint T=t+(lint)B*b;T*=r;
	return (t+(lint)B*b)*r*(R+(lint)G*g);
}
int main()
{
	int n;scanf("%d%d%d%d%d",&n,&R,&G,&B,&t);
	lint ans=0;int now=1,pre=0;
	for(int i=1;i<=n;i++,swap(now,pre))
		for(int g=0;g<=i;g++)
		{
			int b=i-g;dp[now][g]=0;
			if(g) dp[now][g]=max(dp[now][g],dp[pre][g-1]+calc(1,g-1,b));
			if(b) dp[now][g]=max(dp[now][g],dp[pre][g]+calc(1,g,b-1));
			ans=max(ans,dp[now][g]+calc(n-i,g,b,R));
		}
	printf("%lld\n",ans);return 0;
}