POJ3211(01揹包)

題目鏈接如下:http://poj.org/problem?id=3211

題目大意:給你N件衣服,衣服一共可能有M種顏色,接下來的N行包含了每件衣服的顏色及其漂洗所要花費的時間。兩個人一起洗衣服,同時只能洗同一種顏色的衣服,這種顏色的衣服洗完之後才能緊接着洗下一種顏色的衣服,問洗完所有的衣服所要花費的最少時間。

解題思路:對於每種顏色的衣服,可以看成一個01揹包問題,揹包的容量爲漂洗這種衣服的時間和的一半。

#include <iostream>
#include <map>
#include <vector>
#include <string.h>
#define maxn 100005
using namespace std;
string color[11];
map<string,int>mc;//顏色映射 
vector<int>ve[11];//記錄每種顏色的衣服的漂洗時間數組 
int dp[maxn];//dp數組 
int main()
{
	ios::sync_with_stdio(false);
	int M,N;
	while(cin>>M>>N&&(N+M))
	{
		for(int i=0;i<11;i++)ve[i].clear();
		mc.clear();
		for(int i=0;i<M;i++)
		{
			cin>>color[i];
			mc[color[i]]=i;
		}
		
		string str;int num;
		for(int i=0;i<N;i++)
		{
			cin>>num>>str;
			ve[mc[str]].push_back(num);
		}
		
		int ans=0;
		for(int i=0;i<M;i++)
		{
			memset(dp,0,sizeof(dp));
			int len=ve[i].size(),sum=0;
			for(int j=0;j<len;j++)sum+=ve[i][j];
			
			int rong=sum/2;//揹包容量 
			for(int j=0;j<len;j++)
			{
				for(int k=rong;k>=ve[i][j];k--)
				dp[k]=max(dp[k],dp[k-ve[i][j]]+ve[i][j]);
			}
			ans+=max(dp[rong],sum-dp[rong]);//取兩個人洗的時間的最大值 
		}
		cout<<ans<<'\n';
	}
	
}

 

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章