4144: [AMPPZ2014]Petrol

题目链接

题目大意：给定一个n个点、m条边的带权无向图，其中有s个点是加油站。
每辆车都有一个油量上限b，即每次行走距离不能超过b，但在加油站可以补满。
q次询问，每次给出x,y,b，表示出发点是x，终点是y，油量上限为b，且保证x点和y点都是加油站，请回答能否从x走到y

题解：每个点到它最近的加油站是最优的，以所有加油站为起点跑最短路，求出每个点到最近的加油站的距离
若一条边连接的两个点的最近加油站不同，新建一条连接两个加油站的边，权为dis[u]+dis[v]+w，对这些边排序 ，把询问也离线排序，对于每个询问，加入边权小于其限制油量的边，并查集维护连通性即可

Orz题解

我的收获：只保留关键点……神奇建模

#include <cstdio>
#include <cstring>
#include <iostream>
#include <queue>
#include <algorithm>
using namespace std;

#define M 200021
#define INF 0x7fffffff
typedef pair<int,int> pii;

int x,y,z;
int n,m,s,t,Q,cnt;
int c[M],f[M],ans[M];
int dis[M],head[M],vis[M],near[M];

struct que{
    int x,y,d,id;
    bool operator<(const que& b)const{return d<b.d;}
}qu[M];

struct data{
    int a,b,c;
    bool operator<(const data& y)const{return c<y.c;}
}nod[M*4];

struct edge{int to,val,nex;}e[M*4];

priority_queue<pii,vector<pii>,greater<pii> >q;

void add(int u,int v,int w){e[t]=(edge){v,w,head[u]};head[u]=t++;}

int find(int x){return f[x]==x?x:f[x]=find(f[x]);}

void dijkstra(){
    for(int i=1;i<=n;i++) dis[i]=INF;
    for(int i=1;i<=s;i++) dis[c[i]]=0,near[c[i]]=c[i],q.push(make_pair(0,c[i]));
    while(!q.empty()){
        int u=q.top().second;q.pop();
        if(vis[u]) continue;vis[u]=1;
        for(int v,i=head[u];i!=-1;i=e[i].nex){
            if(dis[v=e[i].to]>dis[u]+e[i].val){
                dis[v]=dis[u]+e[i].val;near[v]=near[u];
                q.push(make_pair(dis[v],v));
            }
        }
    }
}

void build(){
    for(int x=1;x<=n;x++){
        for(int v,i=head[x];i!=-1;i=e[i].nex){
            if(near[x]!=near[v=e[i].to])
                nod[++cnt]=(data){near[x],near[v],dis[x]+dis[v]+e[i].val};
        }
    }sort(nod+1,nod+1+cnt);
}

void solve(){
    int now=1;
    for(int i=1;i<=n;i++) f[i]=i;
    for(int i=1;i<=Q;i++){
        while(now<=cnt&&nod[now].c<=qu[i].d){
            int dx=find(nod[now].a),dy=find(nod[now].b);
            if(dx!=dy) f[dx]=dy;
            now++;
        }
        int dx=find(qu[i].x),dy=find(qu[i].y);
        if(dx==dy) ans[qu[i].id]=1;
        else ans[qu[i].id]=0;
    }
}

void work()
{
    solve();
    for(int i=1;i<=Q;i++) if(ans[i]) puts("TAK");
    else puts("NIE");
}

void init()
{
    t=0;memset(head,-1,sizeof(head));
    scanf("%d%d%d",&n,&s,&m);
    for(int i=1;i<=s;i++) scanf("%d",&c[i]);
    for(int i=1;i<=m;i++) scanf("%d%d%d",&x,&y,&z),add(x,y,z),add(y,x,z);
    dijkstra();build();cin>>Q;
    for(int i=1;i<=Q;i++) scanf("%d%d%d",&qu[i].x,&qu[i].y,&qu[i].d),qu[i].id=i;
    sort(qu+1,qu+1+Q);
}

int main()
{
    init();
    work();
    return 0;
}