[codeforces 580D]Kefa and Dishes

n很小，很容易想到状态压缩dp，用dp[set][i]表示结尾是i，用了set里的元素的最大值，枚举set和i来更新下一个元素即可。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<set>
using namespace std;
long long val[20],dp[300000][20];
long long n,m,k;
set<long long> s;
long long max(long long x,long long y){
    return x>y?x:y;
}
long long decode(long long x){
    long long ret=0;
    for(long long i=1;i<=n;i++){
        if(x&1){s.insert(i);ret++;}
        x>>=1;
    }
    return ret;
}
long long r[20][20];
int main(){

    cin>>n>>m>>k;
    for(long long i=1;i<=n;i++)cin>>val[i];
    long long x,a,b;
    memset(r,0,sizeof(r));
    for(long long i=1;i<=k;i++){
        cin>>a>>b>>x;
        r[a][b]=x;
    }
    long long len=1<<n;long long ans=-1;
    memset(dp,-1,sizeof(dp));
    for(long long i=1;i<len;i++){
        s.clear();
        long long num=decode(i);
        for(set<long long>::iterator j=s.begin();j!=s.end();j++){
            long long jj=*j;if(num==1)dp[i][jj]=val[jj];
            if(num==m)ans=max(ans,dp[i][jj]);
            for(long long l=1;l<=n;l++)
                if(s.find(l)==s.end())
                    dp[i|(1<<(l-1))][l]=max(dp[i|(1<<(l-1))][l],dp[i][jj]+r[jj][l]+val[l]);
        }
    }
    cout<<ans<<endl;
    return 0;
}