BZOJ 4289: PA2012 Tax

版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/qq_37960603/article/details/82145649

题目传送门
代码:

#include<bits/stdc++.h>
using namespace std;

typedef long long LL;
typedef pair<LL,int>P;
const int maxn=100000+100;
const int maxm=400000+100;
const LL INF=1e18;

struct Edge{

    int to,len,next;
}edge1[maxm],edge2[maxm*4];
struct Node{

    int u,v,len;
}node[maxm];
int head1[maxn],tot1;
int head2[maxm],tot2;
LL dis[maxm];
bool vis[maxm];
int n,m;
int S,T;

inline bool cmp(Node a,Node b){

    return a.len<b.len;
}

void AddEdge1(int u,int v,int len){

    edge1[tot1].to=v;
    edge1[tot1].len=len;
    edge1[tot1].next=head1[u];
    head1[u]=tot1++;
}

void AddEdge2(int u,int v,int len){

    edge2[tot2].to=v;
    edge2[tot2].len=len;
    edge2[tot2].next=head2[u];
    head2[u]=tot2++;
}

void Dijkstra(){

    dis[S]=0;
    priority_queue<P,vector<P>,greater<P> >que;
    que.push(P(0,S));
    while(!que.empty()){

        P p=que.top();
        que.pop();
        int u=p.second;
        if(u==T) return ;
        if(vis[u]) continue;
        vis[u]=true;
        for(int i=head2[u];i!=-1;i=edge2[i].next){

            Edge e=edge2[i];
            int v=e.to;
            if(!vis[v] && dis[v]>dis[u]+e.len){

                dis[v]=dis[u]+e.len;
                que.push(P(dis[v],v));
            }
        }
    }
}

int main(){

    scanf("%d%d",&n,&m);
    tot1=0;
    for(int i=1;i<=n;i++) head1[i]=-1;
    for(int i=1;i<=m;i++){

        int u,v,len;
        scanf("%d%d%d",&u,&v,&len);
        AddEdge1(u,v,len);AddEdge1(v,u,len);
    }
    tot2=0;
    for(int i=0;i<=2*m+6;i++) head2[i]=-1;
    for(int i=1;i<=n;i++){

        int tot3=0;
        for(int j=head1[i];j!=-1;j=edge1[j].next){

            node[tot3].u=j^1;
            node[tot3].v=j;
            node[tot3].len=edge1[j].len;
            tot3++;
        }
        sort(node,node+tot3,cmp);
        for(int i=0;i<tot3;i++)  AddEdge2(node[i].u,node[i].v,node[i].len);
        for(int i=0;i<tot3-1;i++){

            AddEdge2(node[i].v,node[i+1].v,node[i+1].len-node[i].len);
            AddEdge2(node[i+1].v,node[i].v,0);
        }
    }
    S=tot1++,T=tot1++;
    for(int i=head1[1];i!=-1;i=edge1[i].next) AddEdge2(S,i,edge1[i].len);
    for(int i=head1[n];i!=-1;i=edge1[i].next) AddEdge2(i^1,T,edge1[i^1].len);
    for(int i=0;i<=T;i++) dis[i]=INF,vis[i]=false;
    Dijkstra();
    printf("%lld\n",dis[T]);
}

猜你喜欢

转载自blog.csdn.net/qq_37960603/article/details/82145649