2017沈阳网络赛transaction transaction transaction_hdu6201超级汇点跑SPFA

http://acm.hdu.edu.cn/showproblem.php?pid=6201 

/*巧妙之处：建立超级源点和超级汇点，让多源问题变成单源问题
1.  建立超级源点0和超级汇点n+1。
2.  让超级源点（作为起点）连树上所有的点，边权分别为 -a[i]，表示买书花费；
3.  所有树上所有点连接汇点，边权为a[i]，表示卖书花费，
4.  原本的树上两两点之间的边权设为边权为-w，表示花费w。 
5.  然后spfa跑个最长路，答案为dis[超级汇点n+1]。
*/

#include<bits/stdc++.h>
using namespace std;
const int MAXN = 1e5+10;
const int INF = 0x3f3f3f3f;
struct Edge
{
    int v;
    int cost;
    Edge(int _v = 0, int _cost = 0) : v(_v), cost(_cost) {}
};
vector<Edge> E[MAXN];

void addedge(int u, int v, int w)
{
    E[u].push_back(Edge(v, w));
}
bool vis[MAXN]; //在队列标志
int cnt[MAXN];  //每个点的入队列次数
int dist[MAXN];
bool SPFA(int start, int n)
{
    memset(vis, false, sizeof(vis));
    memset(cnt, 0, sizeof(cnt));
    for (int i = 0; i <= n+1; i++)
        dist[i] = -INF;
    vis[start] = true;
    dist[start] = 0;
    cnt[start] = 1;
    queue<int> que;
    while (!que.empty())
        que.pop();
    que.push(start);
    while (!que.empty())
    {
        int u = que.front();
        que.pop();
        vis[u] = false;
        for (int i = 0; i < E[u].size(); i++)
        {
            int v = E[u][i].v;
            if (dist[v] < dist[u] + E[u][i].cost)
            {
                dist[v] = dist[u] + E[u][i].cost;
                if (!vis[v])
                {
                    vis[v] = true;
                    que.push(v);
                    if (++cnt[v] > n)
                        return false;
                    //cnt[i] 为入队列次数，用来判定是否存在负环回路
                }
            }
        }
    }
    return true;
}
int arr[MAXN],n;
int main()
{
    int T;
    scanf("%d",&T);
    while(T--){
        for(int i=0;i<=n+1;i++){
            E[i].clear();
        }
        scanf("%d",&n);
        int s=0,t=n+1;//超级源点 和 超级汇点
        for(int i=1;i<=n;i++){
            scanf("%d",&arr[i]);
            addedge(s,i,-arr[i]);
            addedge(i,t,arr[i]);
        }
        int u,v,w;
        for(int i=0;i<n-1;i++){
            scanf("%d%d%d",&u,&v,&w);
            addedge(u,v,-w);
            addedge(v,u,-w);//无向图
        }
        SPFA(s,n);
        printf("%d\n",dist[t]);
    }
    return 0;
}