The 2018 ACM-ICPC Chinese Collegiate Programming Contest题解(计蒜课第五场)

这场就补一道题哈哈。
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;
}