题目大意:有n个城市,m条路,每个城市有一个权值a,每访问过一次权值下降为b,一个人从起点出发,访问城市数目小于k,所走总路程长度小于l,求所能获得的权值的和最大。
思路:先做一遍flody(或BFS之类的),求出任意两点之间的距离,然后dp,i表示访问的城市的集合(状态压缩),j表示当前所在城市,那么当前点可扩展的条件就是d[i][j]+当前点j到起点的距离<=l,标记该状态是合法的,之后将该点加入集合中,更新其他点的值。如果当前状态合法,利用优先队列贪心,取k次,更新答案。
#include <cstdio>
#include <cstring>
#include <cmath>
#include <iostream>
#include <queue>
#include <algorithm>
using namespace std;
const int INF = 0x3f3f3f3f;
const int maxn = 17;
int n,m,k,l,a[maxn],b[maxn],T;
int g[maxn][maxn],cases = 0;
int d[1<<(maxn)][maxn];
bool vis[1<<(maxn)];
void Flody(){
for(int k=0;k<=n;k++)
for(int i=0;i<=n;i++)
for(int j=0;j<=n;j++)
g[i][j] = min(g[i][j],g[i][k]+g[k][j]);
return;
}
void TSP(){
memset(vis,0,sizeof vis);
memset(d,INF,sizeof d);
d[1][0] = 0;
int ans = 0,up = (1<<(n+1))-1;
for(int i=0;i<=up;i++){
for(int j=0;j<=n;j++)
if(d[i][j]<=l){
if(d[i][j]+g[j][0]<=l) vis[i] = true;
for(int k=1;k<=n;k++)
d[i|(1<<k)][k] = min(d[i|(1<<k)][k],d[i][j]+g[j][k]);
}
if(vis[i]){
priority_queue<pair<int,int> > q;
for(int j=1;j<=n;j++)
if(i&(1<<j)) q.push(make_pair(a[j],b[j]));
if(q.empty()) continue;
int tmp = 0;
for(int j=1;j<=k;j++){
tmp += q.top().first;
q.push(make_pair(max(q.top().first-q.top().second,0),q.top().second));
q.pop();
}
ans = max(ans,tmp);
}
}
printf("Case %d: %d\n",++cases,ans);
}
int main(){
scanf("%d",&T);
while(T--){
int x,y,z;
scanf("%d%d%d%d",&n,&m,&k,&l);
memset(g,INF,sizeof g);
for(int i=0;i<=n;i++)
g[i][i] = 0;
for(int i=1;i<=n;i++)
scanf("%d",&a[i]);
for(int i=1;i<=n;i++)
scanf("%d",&b[i]);
for(int i=0;i<m;i++){
scanf("%d%d%d",&x,&y,&z);
g[x][y] = g[y][x] = min(g[x][y],z);
}
Flody();
TSP();
}
return 0;
}