2019-2020 ACM-ICPC Brazil Subregional Programming Contest I. Interplanetary（思维+floyed）

传送门

题意

给你$n$，$m$表示$n$个点，$m$条无向带权边，接下来给出$n$个点的权值$v_i$ ，再给出每条无向边的两个端点和权值$w$。再给你$q$，接下来$q$次查询，每次查询的格式为
$a$ $b$ $k$ $t$
查询的是$a$到$b$经过的中间节点的值只能是前$k$小的或者是前$k$大的点的最短路长度，如果不可达输出$-1$。
如果$t=0$表示前$k$小，$t=1$表示前$k$大。

数据范围： 2 ⩽ n ⩽ 400 , 0 ⩽ m ⩽ n ( n − 1 ) , − 1 0 9 ⩽ v i ⩽ 1 0 9 , 1 ⩽ w ⩽ 1 0 3 , 1 ⩽ q ⩽ 1 0 5 , 1 ⩽ a , b , k ⩽ n , t ∈ { 0 , 1 } 2\leqslant n\leqslant 400,0\leqslant m\leqslant n(n-1),-10^9\leqslant {v_i}\leqslant 10^9,1\leqslant w\leqslant 10^3,1\leqslant q\leqslant 10^5,1\leqslant a,b,k\leqslant n,t\in\{0,1\} 2⩽n⩽400,0⩽m⩽n(n−1),−109⩽vi​⩽109,1⩽w⩽103,1⩽q⩽105,1⩽a,b,k⩽n,t∈{0,1}

思路

①一看求最短路以及$n$的范围，很容易想到用类似于$floyed$算法的东西求出答案。但是不可能每次查询都$floyed$一遍，于是考虑将询问离线，按照$k$值进行排序。对于前$k$大和前$k$小分离开分别考虑。
②我们将每个点按照权值从小到大排序，再用一个$rk$数组求出排序后每个位置的数是第几大的数。这样就可以将输入的$k$根据其类型转化成相应的第几小或者第几大。
③以转化成第几小为例，$t=0$时，我们按照权值从小到大将所有点$i$依次作为中间点去更新其他的点的最短距离，同时维护一个查询的指针，如果当前查询的是第$s$小，那么当$s<rk[i]$时，对应查询的两个点当前的最短距离就是答案。$t=1$时类似，留给读者自行思考。
④总时间复杂度为$O(n^3)$。

#include<bits/stdc++.h>
#define ll long long
#define inf 0x3f3f3f3f3f3f3f3fLL
#define rep(i,a,b) for(int i=(a);i<=(b);i++)
#define dep(i,a,b) for(int i=(a);i>=(b);i--)
#define ls (rt<<1)
#define rs (rt<<1|1)
#define fi first
#define se second
#define pb push_back
using namespace std;
const int maxn=4e2+5;
//const double pi=acos(-1.0);
//const double eps=1e-9;
//const ll mo=1e9+7;
int n, m, k;
template <typename T>
inline void read(T &X){
    X=0;int w=0; char ch=0;
    while(!isdigit(ch)) {w|=ch=='-';ch=getchar();}
    while(isdigit(ch)) X=(X<<3)+(X<<1)+(ch^48),ch=getchar();
    if(w) X=-X;
}
struct node{
    int id, val;
    bool operator<(node aa)const{
        return val < aa.val;
    }
}a[maxn];
ll w[maxn][maxn];
int c[maxn],rk[maxn];
struct edge{
    int u, v;
    ll w;
} e[100005];
struct qnode{
    int u, v, w;
    int id;
    bool operator<(qnode qa)const{
         return w < qa.w;
    }
}q[2][100005];
int num[2];
ll ans[100005];
void init(){
    rep(i, 1, n) rep(j, 1, n) w[i][j] = w[j][i] = inf;
    rep(i, 1, n) w[i][i] = 0;
}
void go(int k){
    rep(i, 1, n) rep(j, 1, n) if(w[i][k]!=inf&&w[k][j]!=inf){
        w[i][j] = min(w[i][j], w[i][k] + w[k][j]);
    }
}
void cal0(int id){
    int j = 1;
    rep(i,1,n){
        while(j<=num[id]&&q[id][j].w<rk[i]){
            ll dst = w[q[id][j].u][q[id][j].v];
            ans[q[id][j].id] = (dst == inf ? -1 : dst);
            j++;
        }
        go(i);
    }
    while(j<=num[id]){
        ll dst = w[q[id][j].u][q[id][j].v];
        ans[q[id][j].id] = (dst == inf ? -1 : dst);
        j++;
    }
}
void cal1(int id){
    int j = num[id];
    dep(i,n,1){
        while(j>=1&&q[id][j].w>rk[i]){
            ll dst = w[q[id][j].u][q[id][j].v];
            ans[q[id][j].id] = (dst == inf ? -1 : dst);
            j--;
        }
        go(i);
    }
    while(j>=1){
        ll dst = w[q[id][j].u][q[id][j].v];
        //cout << "debug: " << q[id][j].u << " " << q[id][j].v << " " << dst << endl;
        ans[q[id][j].id] = (dst == inf ? -1 : dst);
        j--;
    }
}
void solve(){
    read(n);
    read(m);
    rep(i,1,n) {
        a[i].id = i;
        read(a[i].val);
    }
    sort(a + 1, a + 1 + n);
    rk[1] = 1;
    rep(i, 1, n) {
        c[a[i].id] = i;
        if(i>1&&a[i].val==a[i-1].val)
            rk[i] = rk[i - 1];
        else if(i>1)
            rk[i] = rk[i - 1] + 1;
    }
    init();
    rep(i,1,m){
        read(e[i].u);
        read(e[i].v);
        read(e[i].w);
        int x = c[e[i].u], y = c[e[i].v];
        w[x][y] = w[y][x] = min(w[x][y], e[i].w);
    }
    read(k);
    rep(i,1,k){
        int u, v, w, id;
        read(u);
        read(v);
        read(w);
        read(id);
        u = c[u];
        v = c[v];
        if(id)
            q[id][++num[id]] = {u, v, rk[n] - w + 1, i};
        else
            q[id][++num[id]] = {u, v, w, i};
    }
    rep(i, 0, 1) sort(q[i] + 1, q[i] + 1 + num[i]);
    cal0(0);

    init();
    rep(i,1,m){
        int x = c[e[i].u], y = c[e[i].v];
        w[x][y] = w[y][x] = min(w[x][y], e[i].w);
    }
    //rep(i, 1, n) rep(j, 1, n) cout << "debug: " << i << ' ' << j << " " << w[i][j] << endl;
    cal1(1);

    rep(i, 1, k) printf("%lld\n", ans[i]);
}
int main(){
    int T=1,cas=1;
    //read(T);
    while(T--){
        solve();
    }
    //system("pause");
    return 0;
}