hdu6201(无向图最长路)

设置一个虚拟起点和虚拟终点,每个点与起点间一条负边,值为这个点书的价值的相反数(代表买书花钱),每个点与终点连一条正边,值为这个点的书的价格(代表卖书赚钱)。然后按照图中给的边建无向边,权值为负(代表路费)。然后就是跑最长路,spfa改一下松弛条件就行

#include<cstdio>
#include<vector>
#include<cstring>
#include<queue>
using namespace std;
const int maxn = 100000+10;
const int inf = 1e9;
int n;
int cost[100000+10];
struct node
{
    int to;
    int w;
    node(int a,int b):to(a),w(b){}
};
vector<node>G[maxn];
int dist[maxn],vis[maxn];
void spfa()
{
    queue<int>q;
    for(int i=0;i<=n+1;i++) dist[i] = -inf;
    memset(vis,0,sizeof(vis));
    q.push(0); vis[0] = 1; dist[0] = 0;
    while(!q.empty())
    {
        int u = q.front(); q.pop(); int len = G[u].size();
        for(int i=0;i<len;i++)
        {
            int v = G[u][i].to;
            int w = G[u][i].w;
            if(dist[v]<dist[u]+w)
            {
                dist[v] = dist[u] + w;
                if(!vis[v]){vis[v] = 1; q.push(v); }
            }
        }
        vis[u] = 0;
    }
    printf("%d\n",dist[n+1]);
}
int main()
{
    int cases,u,v,w;
    scanf("%d",&cases);
    while(cases--)
    {
        scanf("%d",&n);
        for(int i=0;i<n+5;i++) G[i].clear();
        for(int i=1;i<=n;i++) scanf("%d",&cost[i]);
        for(int i=1;i<=n;i++)
        {
            G[0].push_back(node(i,-cost[i]));
            G[i].push_back(node(n+1,cost[i]));
        }
        for(int i=1;i<n;i++)
        {
            scanf("%d%d%d",&u,&v,&w);
            G[u].push_back(node(v,-w));
            G[v].push_back(node(u,-w));
        }
        spfa();
    }
    return 0;
}



评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值