題目鏈接如下: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';
}
}