0-1背包-变形

题目链接：https://ac.nowcoder.com/acm/contest/642/D
题目大意：


思路：一开始想的是，01背包dp出dp[n][s-1]。然后标记选的音乐，从没有标记的音乐找到MAX，输出：dp[n][s-1]+MAX;
然后WA了，
想到一个错误样例：

1
5 7
6 1 1 1 3

方案是：6
那么答案是： 6+3=9

实际上：
可以：选择1 1 1 3=6
答案：6+6=12

正确的思路：从输入的时候找到最大值，跑01背包dp[n][s]，dp时不能包含MAX的音乐，如果dp[n][s]等于s说明必须要去一首音乐，并且可以加上其他的音乐，那么就可以：dp[n][s-1]+MAX;这样就可以得到最大值。

如果dp[n][s]!=s,那么说明有时间来播放最长时间的音乐，输出dp[n][s]+MAX;

#include<bits/stdc++.h>
using namespace std;

int a[210];
int b[210];
int dp[80005];//背包的容量

int main()
{
    int T, n, s;
    scanf("%d",&T);
    while(T--)
    {
        int MAX=0, k=-1;
        scanf("%d%d",&n,&s);
        memset(dp, 0, sizeof(dp));
        for(int i=1;i<=n;i++)
        {
            scanf("%d",&a[i]);
            b[i]=a[i];
            if(MAX<b[i])
            {
                MAX=b[i];
                k=i;
            }
        }


        for(int i=1;i<=n;i++)
        {
            if(i==k)
            {
                continue;
            }
            for(int j=s; j>=a[i]; j--)
            {
                dp[j]=max(dp[j], dp[j-a[i]]+b[i]);
            }
        }
        if(dp[s]==s)
        {
            printf("%d\n",dp[s-1]+MAX);
        }
        else
        {
            printf("%d\n",dp[s]+MAX);
        }
    }

    return 0;
}