POJ_3211 Washing Clothes (01背包)

　　题意是夫妻俩洗衣服，只有一个盆。为了防止不同颜色的衣服混色，他俩一个用一个盆同时洗衣服，但只能是相同颜色的衣服。现在知道颜色数M, 衣服数N， 每件衣服的洗完所用的时间和每件衣服的颜色。求最短多长时间把所有的衣服洗完。

　　思路：把每种颜色的衣服归类，求出第i种颜色衣服所用的总时间sum[i], 以sum[i]/2为背包容量，求出该容量是背包的最大value，sum[i] - f[sum[i]/2]就是洗这种颜色的衣服所用的时间。

　　ps:wa了两次，不知道为什么，又重新敲了一遍就过了。

My Code:

#include <iostream>
#include <cstdio>
#include <cstring>
#include <string>

using namespace std;

const int M = 11;
const int N = 110;

int sum[M];
int num[M];
int w[M][N];
int f[1000*101];
string s[M];
string st;

int main() {
    //freopen("data.in", "r", stdin);

    int n, m, V, i, j, k, ans;
    while(~scanf("%d%d", &m, &n)) {
        if(!m && !n)    break;
        for(i = 0; i < M; i++)     s[i].clear();
        memset(num, 0, sizeof(num));
        memset(sum, 0, sizeof(num));

        for(i = 0; i < m; i++) {
            cin >> s[i];
        }
        for(j = 0; j < n; j++) {
            cin >> V >> st;
            for(i = 0; i < m; i++) {
                if(st == s[i])    {w[i][num[i]++] = V; sum[i] += V; st.clear();}
            }
        }
        ans = 0;
        for(i = 0; i < m; i++) {
            if(sum[i] == 0)    continue;
            memset(f, 0, sizeof(f));
            V = sum[i]/2;
            for(j = 0; j < num[i]; j++) {
                for(k = V; k >= w[i][j]; k--) {
                    f[k] = max(f[k], f[k-w[i][j]] + w[i][j]);
                }
            }
            ans += (sum[i] - f[V]);
        }
        printf("%d\n", ans);
    }
    return 0;
}