codeforces 580D Kefa and Dishes

传送门：http://codeforces.com/problemset/problem/580/d

思路：状压DP，f[i][j]表示最后一个为i，已选取的菜的状态为j。

happiness。

#include<cstdio>
#include<cstring>
#include<algorithm>
const int maxt=540000;
using namespace std;
typedef long long ll; 
int n,m,k,g[20][20],a[20],pow[20];ll f[20][maxt],ans;int main(){pow[0]=1;for (int i=1;i<=19;i++) pow[i]=pow[i-1]<<1;scanf("%d%d%d",&n,&m,&k);for (int i=0;i<n;i++) scanf("%d",&a[i]);for (int i=1,x,y,z;i<=k;i++) scanf("%d%d%d",&x,&y,&z),x--,y--,g[x][y]=z;for (int i=0;i<n;i++) f[i][pow[i]]=a[i];for (int i=1;i<m;i++)for (int s=0;s<pow[n];s++){int cnt=0;for (int j=0;j<n;j++) if (s&pow[j]) cnt++;if (cnt!=i) continue;for (int j=0;j<n;j++)if (!(s&pow[j])){for (int k=0;k<n;k++)if (s&pow[k])f[j][s|pow[j]]=max(f[j][s|pow[j]],f[k][s]+a[j]+g[k][j]);}}for (int s=0;s<pow[n];s++){int cnt=0;for (int j=0;j<n;j++) if (s&pow[j]) cnt++;if (cnt!=m) continue;for (int i=0;i<n;i++) ans=max(ans,f[i][s]);}printf("%I64d\n",ans);return 0;
}