HDU 6201 最长路

题目链接

题意：

一颗树每个点有权值表示此点买入/卖出一本书的价格。还有n-1条边，每条边有一个权值表示经过该边需要的路费。求在树上某个起点买入一本书，某个终点卖出一本书的最大收益为多少。起点和终点可以为同一点。

思路：

为题目加入一个超级源点0和一个超级汇点n+1，源点到任意点权值为p[i]，任意点到汇点权值为-p[i]，连边之间权值为-w，跑一边最长路即可获得答案。

C++代码：

#include<bits/stdc++.h>
using namespace std;
const int maxn = 100010;
const int maxm = 400010;
const int inf  = 0x3f3f3f3f;

int n,tol,head[maxn];
struct edge
{
    int to,cost,next;
}es[maxm];

void addedge( int u , int v , int w )
{
    es[tol].to = v;
    es[tol].cost = w;
    es[tol].next = head[u];
    head[u] = tol++;
}

int dis[maxn],vis[maxn];

struct node
{
    int pos,dis;
    friend bool operator< ( const node&a , const node&b )
    {
        return a.dis<b.dis;
    }
};

void dijkstra()
{
    memset ( dis , -inf , sizeof(dis) );
    memset ( vis , 0 , sizeof(vis) );
    dis[0] = 0;
    node p,q;
    priority_queue<node>Q;
    p.pos = 0;
    p.dis = 0;
    Q.push(p);
    while ( !Q.empty() )
    {
        p = Q.top();
        Q.pop();
        int u = p.pos;
        for ( int i=head[u] ; i!=-1 ; i=es[i].next )
        {
            int v = es[i].to,w = es[i].cost;
            if ( dis[u]+w>dis[v] )
            {
                dis[v] = dis[u]+w;
                p.pos = v;
                p.dis = dis[v];
                Q.push(p);
            }
        }
    }
}

int main()
{
    int T; scanf ( "%d" , &T );
    while ( T-- )
    {
        scanf ( "%d" , &n );
        tol = 0;
        memset ( head , -1 , sizeof(head) );
        for ( int i=1 ; i<=n ; i++ )
        {
            int p; scanf ( "%d" , &p );
            addedge( 0 , i , p );
            addedge( i , n+1 , -p );
        }
        for ( int i=1 ; i< n ; i++ )
        {
            int u,v,w;
            scanf ( "%d%d%d" , &u , &v , &w );
            addedge ( u , v , -w );
            addedge ( v , u , -w );
        }
        dijkstra();
        printf ( "%d\n" , dis[n+1] );
    }
    return 0;
}