[codeforces 580D]Kefa and Dishes

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

#include
#include
#include
#include
using namespace std;
long long val[20],dp[300000][20];
long long n,m,k;
set 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<::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<


你可能感兴趣的:(DP)