CSUST 精灵王国 题解(思维+最短路)

题目链接

题目思路

只有30条路，直接把有铁路的60个点直接全部跑一次最短路，然后问两个点之间的最短距离，要么坐了地铁，那么就是到60个点中的一个最短路加上从这个有铁路的点到另一个点的最短路，要么就是不做地铁，不做地铁一个前缀和就行了。

思路来源

代码

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
typedef pair<int,int> pii;
#define fi first
#define se second
#define debug printf("aaaaaaaaaaa\n");
const int maxn=1e5+100,inf=0x3f3f3f3f,mod=1e9+7;
const ll INF=0x3f3f3f3f3f3f3f3f;
const double eps=1e-7;
int n,m,q;
int d[maxn];
int head[maxn],cnt;
struct edge{
    int to,next,w;
}e[maxn<<2];
ll dis[maxn][65];
ll pre[maxn],sum;
int num[maxn],tot;
struct node{
    ll len;
    int pos,id;
    friend bool operator<(node a,node b){
        return a.len>b.len;
    }
};
void add(int u,int v,int w){
    e[++cnt]={v,head[u],w};
    head[u]=cnt;
}

void dij(int x,int id){
	priority_queue<node > que;
	dis[x][id]=0;
	que.push({0,x,id});
	while(!que.empty()){
		ll len=que.top().len;
		int pos=que.top().pos;
		int id=que.top().id;
		que.pop();
		if(len!=dis[pos][id]) continue;
        for(int i=head[pos];i;i=e[i].next){
            if(dis[e[i].to][id]>dis[pos][id]+e[i].w){
                dis[e[i].to][id]=dis[pos][id]+e[i].w;
                que.push({dis[e[i].to][id],e[i].to,id});//一定注意是len为fi
            }
        }

	}
}
ll cal(int x,int y){
    if(x>y) swap(x,y);
    return min(pre[y]-pre[x],sum-(pre[y]-pre[x]));
}
int main(){
    memset(dis,0x3f,sizeof(dis));
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++){
        scanf("%d",&d[i]);
        add(i,i%n+1,d[i]);
        add(i%n+1,i,d[i]);
        if(i!=n){
            pre[i+1]=pre[i]+d[i];
        }
        sum+=d[i];
    }
    set<int> se;
    for(int i=1,u,v,w;i<=m;i++){
        scanf("%d%d%d",&u,&v,&w);
        add(u,v,w),add(v,u,w);
        se.insert(u);
        se.insert(v);
    }
    for(auto i:se){
        num[++tot]=i;
        dij(i,tot);
    }
    scanf("%d",&q);
    for(int i=1,x,y;i<=q;i++){
        scanf("%d%d",&x,&y);
        ll ans=cal(x,y);
        for(int j=1;j<=tot;j++){
            // x->num[j]->y
            ans=min(ans,cal(num[j],x)+dis[y][j]);
        }
        printf("%lld\n",ans);
    }
    return 0;
}