hdu 6201 transaction transaction transaction

题意：一棵树，有点权，边权，问选择2个点，即abs(a[i]-a[j]) - (这两点经过的边权） 最大

思路： 我们假设买为 x ,卖为y ，路径消耗为z  ，即 x- (y+z)最大，即-(x-(y+z))最小，

　　　 我们增加一个0点，他到每个点的距离为- 该点权值 （即买书的钱

　　　　增加个n+1点，到每个点的距离为该点权值（即卖书的到的钱

　　　　0到n+1跑个最短路即可

#include<bits/stdc++.h>
using namespace std;
const int N=3e6+10;;
const int INF=1000000;
int n,m,dis[N],head[N],len;
bool vis[N];
int b[N];

struct edge
{
    int to,val,next;
}e[N];

void add(int from,int to,int  val)
{
    e[len].to=to;
    e[len].val=val;
    e[len].next=head[from];
    head[from]=len++;
}
struct point
{
    int val,id;
    point(int id,int val):id(id),val(val){}
    bool operator <(const point &x)const{
        return val>x.val;
    }
};
void dijkstra(int s)
{
    memset(vis,0,sizeof(vis));
    for(int i=0;i<N;i++)
        dis[i]=INF;
    priority_queue<point> q;
    q.push(point(s,0));
    dis[s]=0;
    while(!q.empty())
    {
        int cur=q.top().id;
        q.pop();
        if(vis[cur]) continue;
        vis[cur]=true;
        for(int i=head[cur];i!=-1;i=e[i].next)
        {
            int id=e[i].to;
            if(!vis[id] && dis[cur]+e[i].val < dis[id])
            {
                dis[id]=dis[cur]+e[i].val;
                q.push(point(id,dis[id]));
            }
        }
    }
}

int main()
{
    int t;
    cin>>t;
    while(t--)
    {
        scanf("%d",&n);
        len=0;
        for(int i=1;i<=n;i++) scanf("%d",&b[i]);
        memset(head,-1,sizeof(head));
        for(int i=1;i<n;i++)
        {
            int from,to,val;
            scanf("%d%d%d",&from,&to,&val);
            add(from,to,val);
            add(to,from,val);
        }
        for(int i=1;i<=n;i++){
            add(0,i,-b[i]);
            add(i,0,-b[i]);
        }
        for(int i=1;i<=n;i++){
            add(n+1,i,b[i]);
            add(i,n+1,b[i]);
        }
        dijkstra(0);
        printf("%d
",abs(dis[n+1]));
    }
    return 0;

}