Educational Codeforces Round 93 (Rated for Div. 2) D. Colored Rectangles

题意:分别给出三种颜色的木棍,每个木棍有一个长度。每次取两种颜色的木棍分别做一个长方形的长和宽,问最后所得最大的面积之和是多少?

就是一道撒比dp题,被我搞黄了。
首先有个前提是,为了取最大值,每次操作,取两种木棍时,一定取的是当前两种颜色木棍分别的最大长度才能保证答案正确。
也就是说假如
R颜色的木棍最大长度为d,第n(n>=2)大长度为c
G颜色木棍的最大长度为b,第n(n>=2)大长度为a
那么显然db+ca>=bc+da (易证)
那么很显然,就是讲每个颜色的木棍分别排序,然后每次取两个木棍最大值,类似栈的那种操作。
很显然,我们就能通过dp解决了
code:

#include 
#define ll long long
using namespace std;
#define maxn 205
int r[maxn];
int g[maxn];
int b[maxn];
int rl,gl,bl;
int dp[205][205][205];
int main(){
	
	cin>>rl>>gl>>bl;
	for(int i=1;i<=rl;i++)cin>>r[i];
	for(int i=1;i<=gl;i++)cin>>g[i];
	for(int i=1;i<=bl;i++)cin>>b[i];
	sort(r+1,r+rl+1);
	sort(g+1,g+gl+1);
	sort(b+1,b+bl+1);
	for(int i=1;i<=rl;i++)//初始化 
		for(int j=1;j<=gl;j++)
		{
			dp[i][j][0] = max(dp[i][j][0],dp[i-1][j-1][0]+r[i]*g[j]);
		}
	for(int i=1;i<=gl;i++)
		for(int j=1;j<=bl;j++)
		{
			dp[0][i][j] = max(dp[0][i][j],dp[0][i-1][j-1]+g[i]*b[j]);
		}
	for(int i=1;i<=rl;i++)
		for(int j=1;j<=bl;j++)
		{
			dp[i][0][j] = max(dp[i][0][j],dp[i-1][0][j-1]+r[i]*b[j]);
		}
	int ans = 0;
	for(int i=1;i<=rl;i++)//简单的三维dp,也可以写成dfs(rl,gl,bl)。 
		for(int j=1;j<=gl;j++)
			for(int k=1;k<=bl;k++)
			{
				dp[i][j][k] = max(dp[i][j][k],max(dp[i-1][j-1][k]+r[i]*g[j],max(dp[i][j-1][k-1]+g[j]*b[k],dp[i-1][j][k-1]+r[i]*b[k])));
				ans = max(ans,dp[i][j][k]);
			}
	cout<<ans<<endl;
}

你可能感兴趣的:(ACM,算法,动态规划)