#include <bits/stdc++.h>
using namespace std;
/*
题目大意:
给定n个菜,每个菜吃完能增加一个满意度ai,并且有m个吃的规定,每个规定有三个参数x,y,c
如果吃y号菜时上一道菜吃的是x,那么吃完y号菜可以额外增加c的满意度。现在问你,可以吃k个
菜,每个菜只能点一份,满意度最大能是多少。
思路:
定义一个dp[][],第一维为当前菜的状态,1表示吃过了,0表示没吃过;第二维表示当前状态下最
后一个吃的菜。状压dp一下,当当前状态的1的个数为k时,即为答案的一种可能。
题目链接:https://codeforces.com/problemset/problem/580/D
*/
typedef long long ll;
typedef pair<int, int> pii;
const int INF = 0x3f3f3f3f;
const double eps = 1e-6;
const double pi = acos(-1.0);
const int N = 20;
ll a[N];
ll c[N][N];
ll dp[1 << 19][19];
int main()
{
int n, m, tot;
scanf("%d%d%d", &n, &tot, &m);
for(int i = 0; i < n; i++)
{
scanf("%lld", &a[i]);
}
while(m--)
{
int x, y;
scanf("%d%d", &x, &y);
scanf("%lld", &c[x - 1][y - 1]);
}
for(int i = 0; i < n; i++)
{
dp[1 << i][i] = a[i];
}
ll ans = 0;
for(int i = 0; i < (1 << (n + 1)) - 1; i++)
{
int cnt = 0;
for(int j = 0; j < n; j++)
{
if(((i >> j) & 1) == 0) // 最后一个吃的菜为j
{
continue;
}
cnt++;
for(int k = 0; k < n; k++)
{
if(((i >> k) & 1) == 0) // 第k个菜还没吃
{
dp[i | (1 << k)][k] = max(dp[i | (1 << k)][k], dp[i][j] + c[j][k] + a[k]);
}
}
}
if(cnt == tot) // 当前状态已经吃了tot个菜了
{
for(int j = 0; j < n; j++)
{
ans = max(ans, dp[i][j]);
}
}
}
printf("%lld\n", ans);
return 0;
}
CodeForces-580D(状压dp)
最新推荐文章于 2022-08-29 19:55:25 发布