hdu 4610 Cards

http://acm.hdu.edu.cn/showproblem.php?pid=4610

先求出每个数的得分情况，分数和得分状态,(1<<4)种状态

按分数从大到小排序 然后每种状态取一个数(如果有的话)

然后对 dp[i][j] 进行背包 dp[i][j] 表示的是选了i个数选的总状态为j情况下的最大值

然后根据每个 dp[i][j] 对数组剩余的数进行最优选择(在不破坏 最终状态 j 的情况下，尽量选单位分数高的)

最后求最大的情况

代码：

#include<iostream>
#include<cstdio>
#include<string>
#include<cstring>
#include<cmath>
#include<set>
#include<map>
#include<stack>
#include<vector>
#include<algorithm>
#include<queue>
#include<stdexcept>
#include<bitset>
#include<cassert>
#include<deque>
#include<numeric>

using namespace std;

typedef long long ll;
typedef unsigned int uint;
const double eps=1e-12;
const int INF=0x3f3f3f3f;
const ll MOD=1000000007;
const int H=1000005;
const int K=10005;
const int N=1005;
const int M=(1<<4);
struct node
{
    int a,b;
    int point,k;
}in[N];
bool prime(int x)
{
    if(x==1) return false;
    for(int i=2;i*i<=x;++i)
    if(x%i==0)
    return false;
    return true;
}
void get(node &x)
{
    int a=x.a;
    int num=0,sum=0;
    for(int i=1;i*i<=a;++i)
    if(a%i==0)
    {
        ++num;
        sum+=i;
        int j=a/i;
        if(i!=j)
        {++num;sum+=j;}
    }
    x.k=0;x.point=0;
    if(num==2)
    {x.point++;x.k|=1;}
    if(prime(num))
    {x.point++;x.k|=2;}
    if(prime(sum))
    {x.point++;x.k|=4;}
    int h=(int)(sqrt(1.0*a)+0.5);
    int h1=(int)(sqrt(1.0*h)+0.5);
    if(a==1||((num&1)==0&&((num>>1)&1)==0)||((num&1)==1&&h1*h1==h))
    {x.point++;x.k|=8;}
}
bool cmp(node x,node y)
{
    return x.point>y.point;
}
int main()
{
    //freopen("data.in","r",stdin);
    //freopen("1011.in","r",stdin);
    //freopen("1011.out","w",stdout);
    int T;
    scanf("%d",&T);
    while(T--)
    {
        int n,k;
        scanf("%d %d",&n,&k);
        for(int i=0;i<n;++i)
        {
            scanf("%d %d",&in[i].a,&in[i].b);
            get(in[i]);
            if(i>0) printf(" ");
            printf("%d",in[i].point);
        }
        printf("\n");
        int bit[4];
        for(int i=0;i<4;++i)
        scanf("%d",&bit[i]);
        sort(in,in+n,cmp);
        int dp[20][20];
        memset(dp,-1,sizeof(dp));
        dp[0][0]=0;
        bool had[20];
        memset(had,false,sizeof(had));
        for(int i=0;i<n;++i)
        if(had[in[i].k]==false)
        {
            had[in[i].k]=true;
            node &w=in[i];
            for(int i=M-1;i>=0;--i)
            for(int j=0;j<M;++j)
            if(dp[i][j]!=-1)
            {
                int l=i+1;
                int r=(j|w.k);
                dp[l][r]=max(dp[l][r],dp[i][j]+w.point);
            }
            w.b--;
        }
        int ans=-INF;
        for(int i=0;i<=16;++i)
        for(int j=0;j<16;++j)
        if(i<=k&&dp[i][j]!=-1)
        {
            int sum=dp[i][j];
            for(int l=0;l<4;++l)
            if((j&(1<<l))==0)
            sum+=bit[l];
            int w=k-i;
            for(int l=0;w>0&&l<n;++l)
            if((j|in[l].k)==j)
            {
                if(w>=in[l].b)
                {
                    w-=in[l].b;
                    sum+=(in[l].b*in[l].point);
                }else
                {
                    sum+=(w*in[l].point);
                    w=0;
                }
                if(w==0)
                break;
            }
            if(w==0)
            ans=max(ans,sum);
        }
        printf("%d\n",ans);
    }
    return 0;
}