BZOJ 4289: PA2012 Tax

题目传送门
代码：

#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]);
}