luogu1441 砝码称重

题目大意

  现有n个砝码,重量分别为a1,a2,a3,……,an,在去掉m个砝码后,问最多能称量出多少不同的重量(不包括0)。n≤20,m≤4,m<n,ai≤100。

题解

错误做法

  首先枚举子集预处理出不去掉砝码所能得到的重量,随后枚举组合得到去掉的砝码集合A,随后在去掉的砝码里枚举子集得到损失的重量数。

  但是请先试一试样例!这么做是不对的!因为损失的重量还与A内和A外的砝码有关!

  完了完了,赶紧打个暴力,直接枚举A以外的砝码的子集来得到能凑成的重量吧。。。

正确做法

  枚举子集得到重量?优化子集问题为何不想到01背包!

  另外有剪枝:优化搜索顺序,将砝码按重量排序,随后如果枚举组合时后一个元素与前一个元素相同,则跳过。

#include <cstdio>
#include <cstring>
#include <algorithm>
#include <vector>
using namespace std;

const int MAX_N = 25, MAX_W = 2010, INF = 0x3f3f3f3f;
int N, M, TotW, Ans;
int W[MAX_N];
bool F[MAX_W];

void DP()
{
    memset(F, false, sizeof(F));
    F[0] = true;
    for (int i = 1; i <= N; i++)
        for (int j = TotW; j >= W[i]; j--)
            F[j] |= F[j - W[i]];
    int ans = 0;
    for (int i = 1; i <= TotW; i++)
        ans += F[i];
    Ans = max(Ans, ans);
}

void DoSth(vector<int>& chosen)
{
    static int tempW[MAX_N];
    for (int i = 0; i < chosen.size(); i++)
    {
        tempW[chosen[i]] = W[chosen[i]];
        W[chosen[i]] = 0;
    }
    DP();
    for (int i = 0; i < chosen.size(); i++)
        W[chosen[i]] = tempW[chosen[i]];
}

void Dfs(vector<int>& chosen, int n, int m, int begin)
{
    if (m == 0)
    {
        DoSth(chosen);
        return;
    }
    if (n - begin + 1 < m)
        return;
    for (int i = begin; i <= n; i++)
    {
        if (i == begin || W[i] != W[i - 1])
        {
            chosen.push_back(i);
            Dfs(chosen, n, m - 1, i + 1);
            chosen.pop_back();
        }
    }
}

int main()
{
    scanf("%d%d", &N, &M);
    for (int i = 1; i <= N; i++)
        scanf("%d", &W[i]);
    sort(W + 1, W + N + 1);
    for (int i = 1; i <= N; i++)
        TotW += W[i];
    static vector<int> chosen;
    Dfs(chosen, N, M, 1);
    printf("%d\n", Ans);
    return 0;
}

  

posted @ 2018-09-01 10:55  headboy2002  阅读(160)  评论(0编辑  收藏  举报