（最短路+dp）2020牛客寒假算法基础集训营3J.牛牛的宝可梦Go

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

思路：

最短路很简单，floyd即可求出。
然后写了最常见的m²的转移。但是会T掉。化简得方法就是，由于地图很小，所以200步之后可以转移到任意位置，暴力转移前200个，多余前200个就用前缀记录。

代码：

#include<bits/stdc++.h>
#define pii pair<int,int>
#define int long long
#define cl(x,y) memset(x,y,sizeof(x))
#define ct cerr<<"Time elapsed:"<<1.0*clock()/CLOCKS_PER_SEC<<"s.\n";
const int N=1e6+10;
const int mod=1e7+9;
const int maxn=0x3f3f3f3f3f;
const int minn=0xc0c0c0c0;
const int inf=99999999;
using namespace std;
int maze[300][300];
struct mon
{
	int ti,id,w;
	int sum;
	mon()
	{
		sum=0;
	}
}a[N];
int n,m;
void floyd()
{
	int k,i,j;
	for(k=1;k<=n;k++)
		for(i=1;i<=n;i++)
			for(j=1;j<=n;j++)
				maze[i][j]=min(maze[i][j],maze[i][k]+maze[k][j]);
	return;
}
int cmp(mon x,mon y)
{
	return x.ti<y.ti;
}
signed main()
{
	ios::sync_with_stdio(false);
	cin.tie(0);cout.tie(0);
	int i,j;
	cin>>n>>m;
	for(i=1;i<=n;i++)
		for(j=1;j<=n;j++)
			maze[i][j]=i==j?0:maxn;
	for(i=1;i<=m;i++)
	{
		int u,v;
		cin>>u>>v;
		maze[u][v]=1;
		maze[v][u]=1;
	}
	floyd();
	a[0].id=1;a[0].ti=0;a[0].w=0;
	int k;
	cin>>k;
	for(i=1;i<=k;i++)
		cin>>a[i].ti>>a[i].id>>a[i].w;
	sort(a,a+k+1,cmp);
	int res=0,premax;
	for(i=1;i<=k;i++)
	{
		if(i>200)
		{
			premax=max(premax,a[i-200].sum);
			a[i].sum=a[i].w+premax;
		}
		else
			a[i].sum=-maxn;
		for(j=1;j<=min(200LL,i);j++)
		{
			if(a[i].ti-a[i-j].ti>=maze[a[i].id][a[i-j].id])
				a[i].sum=max(a[i].sum,a[i-j].sum+a[i].w);
		}
		res=max(res,a[i].sum);
	}
	cout<<res<<endl;
	return 0;
}