hihocoder1386 Pick Your Players（dp）

Posted 2020-08-11 恶devil魔

题意：

你需要买一个足球队（11个球员），每个球员有位置、价值。花费，有以下限制：
位置分为前锋（1-3人）、中腰（2-5）、后卫（3-5）、守门员（1）
每个人有 value，总的 value 是每个人的value加起来 ，选一个队长，队长的加两次
每个人有个 cost，总花费不能超过给定值
求：最大的 value，相应的最小的 cost，相应的购买方案数（大于1e9输出1e9）
10组数据，500个候选人，value 和 cost：V and C (0 <= V <= 1000, 0 <= C <= 1000)，花费上界：1000

思路：

考虑dp(i,cost,j,k,r,w)=(value,way)表示考虑前i个球员，

费用和为cost，选了j个前锋k个中腰r个后卫w个守门员这个状态，价值和是value,方案数是way。

然后按照dp字面意思转移就可以了。然后第一维的iii这里是为了看起来方便，其实是不用开的。
为了方便处理队长，把球员们按照价值从大到小排序，挑的第一个人当队长就可以了。

 

以上取自ICPCCAMP题解

 

然后我就照着写了一发，怎么调都wa，感觉对对的- -

先存着吧，等以后或许有题解了再对比一下

可以把set换成vector+bool矩阵少一个log复杂度~

有哪位聚聚看出来问题请通知我~感激不尽

/* ***********************************************
Author        :devil
************************************************ */
#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
#include <vector>
#include <queue>
#include <set>
#include <stack>
#include <map>
#include <string>
#include <cmath>
#include <stdlib.h>
#define inf 0x3f3f3f3f
#define LL long long
#define rep(i,a,b) for(int i=a;i<=b;i++)
#define dep(i,a,b) for(int i=a;i>=b;i--)
#define ou(a) printf("%d\\n",a)
#define pb push_back
#define mkp make_pair
template<class T>inline void rd(T &x)
{
    char c=getchar();
    x=0;
    while(!isdigit(c))c=getchar();
    while(isdigit(c))
    {
        x=x*10+c-\'0\';
        c=getchar();
    }
}
#define IN freopen("in.txt","r",stdin);
#define OUT freopen("out.txt","w",stdout);
using namespace std;
const int mod=1e9;
const int N=1e3+10;
pair<int,LL> dp[N][4][6][6][2];
struct wq
{
    char s[12];
    int val,cost;
} a[510];
bool cmp(wq a,wq b)
{
    return a.val>b.val;
}
set<int>eg[1500];
int main()
{
    int t,n,all;
    rd(t);
    while(t--)
    {
        rd(n);
        rep(i,1,n) scanf("%s",a[i].s),rd(a[i].val),rd(a[i].cost);
        rd(all);
        sort(a+1,a+n+1,cmp);
        memset(dp,0,sizeof(dp));
        int ansval=0,anscost=0;
        rep(i,0,1400) eg[i].clear();
        LL anssum=0;
        rep(i,1,n)
        {
            if(a[i].s[0]==\'F\')
            {
                dep(f,3,1)dep(m,5,0)dep(d,5,0)dep(g,1,0)
                {
                    int hs=(f-1)*100+m*10+d+g*1000;
                    if(f+m+d+g>11||eg[hs].size()==0) continue;
                    for(auto j:eg[hs])
                    {
                        int c=j+a[i].cost;
                        if(c>all) continue;
                        if(dp[c][f][m][d][g].first<dp[c-a[i].cost][f-1][m][d][g].first+a[i].val)
                        {
                            dp[c][f][m][d][g].first=dp[c-a[i].cost][f-1][m][d][g].first+a[i].val;
                            dp[c][f][m][d][g].second=dp[c-a[i].cost][f-1][m][d][g].second;
                            eg[hs+100].insert(c);
                        }
                        else if(dp[c][f][m][d][g].first==dp[c-a[i].cost][f-1][m][d][g].first+a[i].val)
                        {
                            dp[c][f][m][d][g].second+=dp[c-a[i].cost][f-1][m][d][g].second;
                            if(dp[c][f][m][d][g].second>mod) dp[c][f][m][d][g].second=mod;
                            eg[hs+100].insert(c);
                        }
                    }
                }
                if(a[i].val*2>dp[a[i].cost][1][0][0][0].first)
                {
                    dp[a[i].cost][1][0][0][0].first=a[i].val*2;
                    dp[a[i].cost][1][0][0][0].second=1;
                    eg[100].insert(a[i].cost);
                }
                else if(a[i].val*2==dp[a[i].cost][1][0][0][0].first)
                {
                    dp[a[i].cost][1][0][0][0].second++;
                    eg[100].insert(a[i].cost);
                }
            }
            else if(a[i].s[0]==\'M\')
            {
                dep(m,5,1)dep(f,3,0)dep(d,5,0)dep(g,1,0)
                {
                    int hs=f*100+(m-1)*10+d+g*1000;
                    if(f+m+d+g>11||eg[hs].size()==0) continue;
                    for(auto j:eg[hs])
                    {
                        int c=j+a[i].cost;
                        if(c>all) continue;
                        if(dp[c][f][m][d][g].first<dp[c-a[i].cost][f][m-1][d][g].first+a[i].val)
                        {
                            dp[c][f][m][d][g].first=dp[c-a[i].cost][f][m-1][d][g].first+a[i].val;
                            dp[c][f][m][d][g].second=dp[c-a[i].cost][f][m-1][d][g].second;
                            eg[hs+10].insert(c);
                        }
                        else if(dp[c][f][m][d][g].first==dp[c-a[i].cost][f][m-1][d][g].first+a[i].val)
                        {
                            dp[c][f][m][d][g].second+=dp[c-a[i].cost][f][m-1][d][g].second;
                            if(dp[c][f][m][d][g].second>mod) dp[c][f][m][d][g].second=mod;
                            eg[hs+10].insert(c);
                        }
                    }
                }
                if(a[i].val*2>dp[a[i].cost][0][1][0][0].first)
                {
                    dp[a[i].cost][0][1][0][0].first=a[i].val*2;
                    dp[a[i].cost][0][1][0][0].second=1;
                    eg[10].insert(a[i].cost);
                }
                else if(a[i].val*2==dp[a[i].cost][0][1][0][0].first)
                {
                    dp[a[i].cost][0][1][0][0].second++;
                    eg[10].insert(a[i].cost);
                }
            }
            else if(a[i].s[0]==\'D\')
            {
                dep(d,5,1)dep(f,3,0)dep(m,5,0)dep(g,1,0)
                {
                    int hs=f*100+m*10+d-1+g*1000;
                    if(f+m+d+g>11||eg[hs].size()==0) continue;
                    for(auto j:eg[hs])
                    {
                        int c=j+a[i].cost;
                        if(c>all) continue;
                        if(dp[c][f][m][d][g].first<dp[c-a[i].cost][f][m][d-1][g].first+a[i].val)
                        {
                            dp[c][f][m][d][g].first=dp[c-a[i].cost][f][m][d-1][g].first+a[i].val;
                            dp[c][f][m][d][g].second=dp[c-a[i].cost][f][m][d-1][g].second;
                            eg[hs+1].insert(c);
                        }
                        else if(dp[c][f][m][d][g].first==dp[c-a[i].cost][f][m][d-1][g].first+a[i].val)
                        {
                            dp[c][f][m][d][g].second+=dp[c-a[i].cost][f][m][d-1][g].second;
                            if(dp[c][f][m][d][g].second>mod) dp[c][f][m][d][g].second=mod;
                            eg[hs+1].insert(c);
                        }
                    }
                }
                if(a[i].val*2>dp[a[i].cost][0][0][1][0].first)
                {
                    dp[a[i].cost][0][0][1][0].first=a[i].val*2;
                    dp[a[i].cost][0][0][1][0].second=1;
                    eg[1].insert(a[i].cost);
                }
                else if(a[i].val*2==dp[a[i].cost][0][0][1][0].first)
                {
                    dp[a[i].cost][0][0][1][0].second++;
                    eg[1].insert(a[i].cost);
                }
            }
            else
            {
                dep(f,3,0)dep(m,5,0)dep(d,5,0)
                {
                    int hs=f*100+m*10+d;
                    if(f+m+d+1>11||eg[hs].size()==0) continue;
                    for(auto j:eg[hs])
                    {
                        int c=j+a[i].cost;
                        if(c>all) continue;
                        if(dp[c][f][m][d][1].first<dp[c-a[i].cost][f][m][d][0].first+a[i].val)
                        {
                            dp[c][f][m][d][1].first=dp[c-a[i].cost][f][m][d][0].first+a[i].val;
                            dp[c][f][m][d][1].second=dp[c-a[i].cost][f][m][d][0].second;
                            eg[hs+1000].insert(c);
                        }
                        else if(dp[c][f][m][d][1].first==dp[c-a[i].cost][f][m][d][0].first+a[i].val)
                        {
                            dp[c][f][m][d][1].second+=dp[c-a[i].cost][f][m][d][0].second;
                            if(dp[c][f][m][d][1].second>mod) dp[c][f][m][d][1].second=mod;
                            eg[hs+1000].insert(c);
                        }
                    }
                }
                if(a[i].val*2>dp[a[i].cost][0][0][0][1].first)
                {
                    dp[a[i].cost][0][0][0][1].first=a[i].val*2;
                    dp[a[i].cost][0][0][0][1].second=1;
                    eg[1000].insert(a[i].cost);
                }
                else if(a[i].val*2==dp[a[i].cost][0][0][0][1].first)
                {
                    dp[a[i].cost][0][0][0][1].second++;
                    eg[1000].insert(a[i].cost);
                }
            }
            //printf("%d\\n",i);
            //rep(f,0,3)rep(m,0,5)rep(d,0,5)rep(g,0,1)rep(c,0,all) if(dp[c][f][m][d][g].second) printf("%d %c %d %d %d %d %d %d %lld\\n",i,a[i].s[0],f,m,d,g,c,dp[c][f][m][d][g].first,dp[c][f][m][d][g].second);
            //system("pause");
        }
        rep(f,1,3)rep(m,2,5)rep(d,3,5)
        {
            if(f+m+d<10) continue;
            if(f+m+d>10) break;
            int hs=f*100+m*10+d+1000;
            //printf("%d %d\\n",hs,eg[hs].size());
            if(!eg[hs].size()) continue;
            for(auto j:eg[hs])
            {
                int c=j;
                //printf("%d %d %lld\\n",c,dp[c][f][m][d][1].first,dp[c][f][m][d][1].second);
                if(dp[c][f][m][d][1].first>ansval)
                {
                    ansval=dp[c][f][m][d][1].first;
                    anscost=c;
                    anssum=dp[c][f][m][d][1].second;
                }
                else if(dp[c][f][m][d][1].first==ansval)
                {
                    if(c<anscost)
                    {
                        anscost=c;
                        anssum=dp[c][f][m][d][1].second;
                    }
                    else if(c==anscost)
                    {
                        anssum+=dp[c][f][m][d][1].second;
                        if(anssum>mod) anssum=mod;
                    }
                }
            }
        }
        if(anssum>mod) anssum=mod;
        printf("%d %d %lld\\n",ansval,anscost,anssum);
    }
    return 0;
}

View Code