2020牛客寒假算法基礎集訓營3 J-牛牛的寶可夢Go(DP+Floyd)

連接：https://ac.nowcoder.com/acm/contest/3004/J

思路：

先用flyod預處理出dis[i][j]表示從i到j點的最短距離

以後開始DP，dp[i]表示出現i個精靈可收穫的最大戰鬥力

那麼dp[i]得轉移方程爲dp[i]=max(dp[i],dp[i-j]+a[i].val) 表示在i以前j個出現的寶可夢的編號

那麼這樣的時間複雜度就會爲K²，可是咱們能夠發現，圖上任意兩點之間的距離都小於等於200，因此j最大隻能爲200，所以複雜度就降爲O(200*K)

注意還要判斷兩個寶可夢出現的時間差要大於等於兩點之間的距離

#include<iostream>
#include<algorithm>
#include<cstring>
 using namespace std;
 typedef long long ll;
 const int maxn=1e5+10;
 const int inf=0x3f3f3f3f;
 int dis[205][205],n,m,u,v,k;
 ll ans=0,dp[maxn];
 struct pok{
     int t,pos,v;
 }a[maxn];
 int cmp(pok a,pok b){return a.t<b.t;}
 void floyd()
 {
      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 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;
     }
    floyd();
    scanf("%d",&k);
    for(int i=1;i<=k;i++)
        scanf("%d%d%d",&a[i].t,&a[i].pos,&a[i].v);
    sort(a+1,a+1+k,cmp);
    a[0].t=0,a[0].pos=1,a[0].v=0;
    for(int i=1;i<=k;i++){
        dp[i]=-inf;
        for(int j=1;j<=200&&i>=j;j++){
            if(a[i].t-a[i-j].t>=dis[a[i].pos][a[i-j].pos])
                dp[i]=max(dp[i],dp[i-j]+a[i].v);
        }
        ans=max(dp[i],ans);
    }
    cout<<ans<<endl;
    return 0;
}