先floyd求一遍最短路。
把每只宝可梦按出现时间的先后顺序排序。
dp[i]表示前i只宝可梦能够捕获的最大战斗力之和。
考虑如何转移,最先想到的一定是k方转移,但是铁定超时。然后发现地图最多只有200个点,也就是说最多200步,就能从一个点走到任意一个点(只要连通的话)。所以每次只用暴力往前for200个点,200以上的记录一下前缀max,每次直接从这里转移。
#include <cstdio>
#include <cstring>
#include <cmath>
#include <algorithm>
#include <iostream>
#include <map>
#include <vector>
#include <stack>
#include <queue>
#include <string>
using namespace std;
typedef long long ll;
typedef pair<int,int> pii;
const int maxn = 1e5+5;
const int inf = 0x3f3f3f3f;
const ll INF = 0x7f7f7f7f7f7f7f7f;
const int mod = 1e9+7;
int dis[205][205];
ll dp[maxn];
struct node
{
int t,p;
ll v;
}a[maxn];
bool cmp(const node &a,const node &b)
{
return a.t<b.t;
}
int main()
{
int n,m;
scanf("%d%d",&n,&m);
memset(dis,0x3f,sizeof(dis));
for(int i=1; i<=m; i++)
{
dis[i][i] = 0;
int u,v;
scanf("%d%d",&u,&v);
dis[u][v] = 1;
dis[v][u] = 1;
}
for(int k=1; k<=n; k++)
{
for(int i=1; i<=n; i++)
{
for(int j=1; j<=n; j++)
{
dis[i][j] = min(dis[i][j],dis[i][k]+dis[k][j]);
}
}
}
int N;
scanf("%d",&N);
for(int i=1; i<=N; i++)
{
scanf("%d%d%lld",&a[i].t,&a[i].p,&a[i].v);
}
sort(a+1,a+1+N,cmp);
ll ans = 0,tmp = 0;
a[0].p = 1;//起点
for(int i=1; i<=N; i++)
{
if(i>200)
{
tmp = max(tmp,dp[i-200]);
dp[i] = a[i].v + tmp;
}
else
{
dp[i] = -INF;
}
for(int j=1; j<=200 && i-j>=0; j++)
{
if(a[i].t - a[i-j].t >= dis[a[i].p][a[i-j].p])
dp[i] = max(dp[i],dp[i-j]+a[i].v);
}
ans = max(ans,dp[i]);
}
printf("%lld",ans);
}
对于为什么一开始要把dp[i]设置成负无穷呢?
因为规定起点一定是1号点,当dp[0] = 0,dp[i]等于负无穷时,能保证起点一定是1号点。可以假设一开始所有dp值都为0,那么转移时会出现从其他点开始(作为起点)转移。