水题
读进来了以后跑一个dijkstra,但是不能再松弛的时候记录pre因为那只是一条最短路
而是重新枚举看是不是dis【x】=dis[y]+w
然后跑一个最大流就行了。
#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cmath>
#include<cstring>
#include<queue>
using namespace std;
typedef int INT;
#define int long long
const int N=500000;
const int INF=1e17+7;
int n,m;
int S=0;
int T;
struct Front_star{
int u,v,w,nxt;
}e[500000],edge[500000];
struct Node{
int u,w;
};
int cnt=0;
int edge_cnt=1;
int first[N]={0};
int head[N]={0};
void insert(int u,int v,int w){
cnt++;
e[cnt].u=u;
e[cnt].w=w;
e[cnt].v=v;
e[cnt].nxt=first[u];
first[u]=cnt;
}
void addedge(int u,int v,int w){
edge_cnt++;
edge[edge_cnt].u=u;
edge[edge_cnt].v=v;
edge[edge_cnt].w=w;
edge[edge_cnt].nxt=head[u];
head[u]=edge_cnt;
}
void add(int u,int v,int w){
addedge(u,v,w);
addedge(v,u,0);
}
int dis[N]={0};
int pre[N]={0};
priority_queue<Node> q;
bool operator <(Node A,Node B){
return A.w<B.w;
}
void dijkstra(int start){
q.push((Node){start,0});
memset(dis,63,sizeof(dis));
memset(pre,0,sizeof(pre));
dis[start]=0;
while(!q.empty()){
Node tmp=q.top();
int x=tmp.u;
q.pop();
for(int i=first[x];i;i=e[i].nxt){
int v=e[i].v;
if(dis[v]>dis[x]+e[i].w){
dis[v]=dis[x]+e[i].w;
q.push((Node){v,tmp.w+e[i].w});
}
}
}
}
//
int d[N]={0};
queue<int> que;
bool bfs(){
memset(d,-1,sizeof(d));
d[S]=1;
que.push(S);
while(!que.empty()){
int x=que.front();
que.pop();
for(int i=head[x];i;i=edge[i].nxt){
int v=edge[i].v;
if(edge[i].w&&d[v]==-1){
d[v]=d[x]+1;
que.push(v);
}
}
}
return d[T]!=-1;
}
int dfs(int st,int ed,int nowdat){
int dat=0;
if(st==ed){
return nowdat;
}
for(int i=head[st];i;i=edge[i].nxt){
int v=edge[i].v;
if(edge[i].w&&d[v]==d[st]+1){
int now=min(nowdat-dat,edge[i].w);
now=dfs(v,ed,now);
edge[i].w-=now;
edge[i^1].w+=now;
dat+=now;
if(nowdat==dat){
return dat;
}
}
}
if(!dat){
d[st]=-2;
}
return dat;
}
void Dinic(){
int s=0;
while(bfs()){
s+=dfs(S,T,INF);
}
cout<<s;
}
INT main(){
// freopen("network.in","r",stdin);
// freopen("network.out","w",stdout);
scanf("%lld%lld",&n,&m);
// cout<<n<<" "<<m<<endl;
for(int i=1;i<=m;i++){
int u,v,w;
scanf("%lld%lld%lld",&u,&v,&w);
insert(u,v,w);
insert(v,u,w);
}
dijkstra(1);
// for(int i=1;i<=n;i++){
// cout<<dis[i]<<" ";
// }
// cout<<'\n';
for(int u=1;u<=n;u++){
for(int i=first[u];i;i=e[i].nxt){
int v=e[i].v;
if(dis[u]+e[i].w==dis[v]){
add(u*2,v*2-1,INF);
// cout<<u<<" "<<v<<endl;
}
}
}
for(int i=1;i<=n;i++){
int dat;
scanf("%lld",&dat);
if(i==1||i==n){
add(i*2-1,i*2,INF);
}
else{
add(i*2-1,i*2,dat);
}
}
T=n*4+10;
add(S,1,INF);
add(n*2,T,INF);
Dinic();
return 0;
}