Codeforces ----- Kefa and Dishes [状压dp]

题目传送门:580D

题目大意:给你n道菜以及每道菜一个权值,k个条件,即第y道菜在第x道后马上吃有z的附加值,求从中取m道菜的最大权值

看到这道题,我们会想到去枚举,但是很显然这是会超时的,再一看数据范围,n只有18,那么我们就可以用状压去做了,dp数组也还是比较好定义的,dp[i][state]表示现在吃到第i道菜状态为state的最大权值,既然有k个限制条件我们就按每个菜去扩展,然后就是一个裸的状压dp了,记得要开long long

#include<iostream>
#include<cstdio>
#include<cmath>
#include<string>
#include<cstring>
#define in(i) (i=read())
using namespace std;
typedef long long lol;
lol read()
{
    lol ans=,f=;
    char i=getchar();
    while(i<''||i>'')
    {
        if(i=='-') f=-;
        i=getchar();
    }
    while(i>=''&&i<='')
    {
        ans=(ans<<)+(ans<<)+i-'';
        i=getchar();
    }
    return ans*f;
}
lol dp[][<<];
lol a[],b[][];
lol n,m,k;
int main()
{
    lol ans=;
    in(n);in(m);in(k);
    for(lol i=;i<=n;i++) in(a[i]);
    for(lol i=;i<=k;i++)
    {
        lol u,v,c;
        in(u);in(v);in(c);
        b[u][v]=c;
    }
    for(lol i=;i<=n;i++) dp[i][<<i-]=a[i];
    for(lol i=;i<(<<n);i++)
    {
        lol cnt=;
        for(lol j=;j<=n;j++)
        {
            if(i&(<<(j-)))
            {
                cnt++;//这里的cnt是已经经过前面的if的语句,也就是说这里的cnt是看此状态有多少位是1,也就是会吃多少道菜,所以才有了下面的if(cnt==m)这个语句
                for(lol k=;k<=n;k++)
                {
                    if(!(i&(<<k-)))
                    {
                        dp[k][i|(<<k-)]=max(dp[k][i|(<<k-)],dp[j][i]+a[k]+b[j][k]);
                    }
                }
            }
        }
        if(cnt==m)
        {
            for(lol j=;j<=n;j++)
                if(i&(<<(j-)))//一定要记得判断
                    ans=max(ans,dp[j][i]);
        }
    }
    cout<<ans<<endl;
    return ;
}