2020牛客寒假算法基础集训营3 J题牛牛的宝可梦Go

#include<bits/stdc++.h>
using namespace std;
const int MAXN=100005;
const int MAXP=505;
const long long INF=1LL<<60;
long long premax,dp[MAXN],ans,dis[MAXP][MAXP];
int n,m,N,u,v;
struct node
{
    int p,t;
    long long val;
}a[MAXN];
int main()
{
    scanf("%d %d",&n,&m);
    for(int i=1;i<=n;++i)
    {
        for(int j=1;j<=n;++j)
        {
            dis[i][j]=i==j?0:INF;
        }
    }
    for(int i=1;i<=m;++i)
    {
        scanf("%d %d",&u,&v);
        dis[u][v]=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]);
            }
        }
    }
    scanf("%d",&N);
    for(int i=1;i<=N;++i)
    {
        scanf("%d %d %lld",&a[i].t,&a[i].p,&a[i].val);
    }
    sort(a+1,a+1+N,[](const node &A,const node &B){return A.t<B.t;});
    a[0].p=1;
    for(int i=1;i<=N;++i)
    {
        if(i>200)
        {
            premax=max(premax,dp[i-200]);
            dp[i]=a[i].val+premax;
        }
        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].val);
            }
        }
        ans=max(ans,dp[i]);
    }
    printf("%lld\n",ans);
    return 0;
}