HYSBZ 1151 动物园zoo - 状压dp

题目描述

题目大意：

N个动物围成一个环，有K个小朋友，每个小朋友可以看到五个连续的动物，每个小朋友都有自己喜欢或讨厌的动物，当有一个自己讨厌的动物被移走或能看到一个自己喜欢的动物时，这个小朋友就会高兴。求最优的移走动物的方案，使得最多的小朋友高兴。

分析：

dp[i][S]：处理到第i个动物，i~i-3的动物移动与否的状态为S。
dp[i+1][S]=max{dp[i+1][newS]，dp[i][S]+Influence} （有移动第i+1个或者不移动两种选择）
Influence：在newS下开心的children数量

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
#include<vector>
#define MAXN 10000
#define MAXC 50000
#define MAXS 32
const int S=(1<<4)-1;

int n,c,fear[100],love[100],dp[MAXC+10][MAXS*2+10],happy[MAXC+10][MAXS+10],ans;

void prepare(int st)
{
    bool p,q;
    for(int s=0;s<=(1<<5)-1;s++){
        p=q=false;
        for(int j=1;j<=fear[0];j++)
            if(!(s&(1<<fear[j]))){
                p=true;
                break;
            }
        for(int j=1;j<=love[0];j++)
            if(s&(1<<love[j])){
                q=true;
                break;
            }
        if(p||q)
            happy[st][s]++;
    }
} // happy[start_pos][start_pos~5->s]
void read()
{
    int pos;
    scanf("%d%d",&n,&c);
    for(int i=1;i<=c;i++){
        scanf("%d%d%d",&pos,&fear[0],&love[0]);
        for(int j=1;j<=fear[0];j++){
            scanf("%d",&fear[j]);
            fear[j]=(fear[j]-pos+n)%n;
        }
        for(int j=1;j<=love[0];j++){
            scanf("%d",&love[j]);
            love[j]=(love[j]-pos+n)%n;
        }
        prepare(pos);
    }
}
void DP(int st)
{
    memset(dp,-1,sizeof dp);
    dp[4][st]=0;
    for(int i=4;i<n;i++){
        for(int s=0;s<=S;s++){
            if(dp[i][s]==-1)
                continue;
            for(int p=0,k;p<=1;p++){
                k=((s&(S^1))>>1)|(p*(1<<3));
                dp[i+1][k]=max(dp[i+1][k],dp[i][s]+happy[i-3][s|(p*(1<<4))]);
            }
        }
    }
    for(int s=0,k;s<=S;s++){
        k=s|((st&1)*(1<<4));
        dp[n][s]+=happy[n-3][k];
        k=((s&(S^1))>>1)|((st&3)*(1<<3));
        dp[n][s]+=happy[n-2][k];
        k=((s&(S^3))>>2)|((st&7)*(1<<2));
        dp[n][s]+=happy[n-1][k];
        k=((s&(1<<3))>>3)|(st<<1);
        dp[n][s]+=happy[n][k];
        ans=max(ans,dp[n][s]);
    }
}
void workout()
{
    for(int s=0;s<=S;s++)
        DP(s);
    printf("%d\n",ans);
}
int main()
{
    read();
    workout();
    return 0;
}