这场就补一道题哈哈。
F. Moving On
n只有200,但是询问有2e4,可以发现其实有很多询问虽然问的不一样但实质是一样的,所以我第一想法是将r离散化,然后将询问离线,对每个询问按照w对应离散化后的r的下标进行分类,对于同一类别的询问跑floyd,然而还是t了。。
之后想到了可以将点按照r的大小排个序,这样就可以了,然而我离散化unique函数返回值一直都搞错了,所以wa了,赛后把它减一就过了。。也算是一个教训吧。
#include<bits/stdc++.h>
using namespace std;
const int inf=0x3f3f3f3f;
struct node
{int u,v;int idx;node(int _u,int _v,int _idx):u(_u),v(_v),idx(_idx) {}node() {}
};
vector<node>V[205];
int r[205];
int d[205][205];
int Map[205][205];
int n,q;
int Wx[205];
int IDX[205];
int k;
void floyd(int w)
{//for(int j=1;j<=n;j++)cout<<IDX[j]<<endl;//memset(d,inf,sizeof(d));for(;k<=n; k++){int go=IDX[k];if(r[go]>w)break;for(int i=1; i<=n; i++)for(int j=1; j<=n; j++){d[i][j]=min(d[i][j],d[i][go]+d[go][j]);}}
}
int ans[20005];
bool cmp(int a,int b)
{return r[a]<r[b];
}
int main()
{int T;scanf("%d",&T);int cas=1;while(T--){k=1;scanf("%d %d",&n,&q);for(int i=1; i<=n; i++)scanf("%d",&r[i]),Wx[i]=r[i],IDX[i]=i;sort(IDX+1,IDX+1+n,cmp);sort(Wx+1,Wx+1+n);/*int wnum=0;for(int i=1;i<=n;i++)if(Wx[i]!=Wx[wnum]){wnum++;Wx[wnum]=Wx[i];}*/int wnum=unique(Wx+1,Wx+1+n)-Wx;//cout<<Wx[wnum]<<endl;wnum--;for(int i=1; i<=n; i++)for(int j=1; j<=n; j++)scanf("%d",&Map[i][j]);for(int i=1; i<=wnum; i++)V[i].clear();int u,v,w;for(int i=1; i<=q; i++){scanf("%d %d %d",&u,&v,&w);int theidx=upper_bound(Wx+1,Wx+1+wnum,w)-Wx;theidx--;V[theidx].push_back(node(u,v,i));}for(int i=1;i<=n;i++)for(int j=1;j<=n;j++)d[i][j]=Map[i][j];for(int i=0; i<=wnum; i++)if(V[i].size()){floyd(Wx[i]);for(int j=0; j<V[i].size(); j++){int u=V[i][j].u,v=V[i][j].v,idx=V[i][j].idx;ans[idx]=d[u][v];}}printf("Case #%d:\n",cas++);for(int i=1; i<=q; i++)printf("%d\n",ans[i]);}return 0;
}