1. 程式人生 > >【luogu P1073 最優貿易】 題解

【luogu P1073 最優貿易】 題解

兩張 turn emp esp 下一條 for cst print scanf

題目鏈接:https://www.luogu.org/problemnew/show/P1073
對於狀態量相互影響的題目,分層圖是個不錯的想法。
考慮在題目中分為:
不交易:
直接從1到n出去,為0
交易:
先在某點買入,再從該點後所在路徑上賣出。
買入賣出是兩個操作,考慮可以分開在兩張圖上做,於是就有了分層圖,共三張圖。
我們把原圖中的路徑都設邊權為0,表示在這條路上走對交易利潤無影響,在第一張圖上買入後,我們就走到下一張圖,準備賣出操作。
設u—>v
所以若從u點買入,到下一條邊的v,即v+n,邊權為買入的花費,-val[u]。
這時我們再第二張圖上的所走,就能保證是再走的路徑是該點往後可以經過的路徑。

這時我們再考慮轉移賣出的情況。
此時已經在v+n—>w+n上
即若在v點賣出,往後可走到w點,所以是v+n到w+2n的一條邊權為val[v]的路徑。
圖建好後,SPFA即可。
技術分享圖片

#include <queue>
#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
using namespace std;
const int maxn = 500010;
int n, m, val[maxn], dis[maxn];
bool vis[maxn];
struct edge{
    int from, to, next, len;
}e[maxn<<2];
int head[maxn], cnt;
queue<int> q;
void add(int u, int v, int w)
{
    e[++cnt].from = u;
    e[cnt].len = w;
    e[cnt].next = head[u];
    e[cnt].to = v;
    head[u] = cnt;
}
void SPFA()
{
    while(!q.empty())
    {
        int now = q.front(); q.pop();
        vis[now] = 0;
        for(int i = head[now]; i != -1; i = e[i].next)
        {
            if(dis[e[i].to] < dis[now] + e[i].len)
            {
                dis[e[i].to] = dis[now] + e[i].len;
                if(!vis[e[i].to])
                {
                    q.push(e[i].to);
                    vis[e[i].to] = 1;
                }
            }
        }
    }
}
int main()
{
    memset(head, -1, sizeof(head));
    scanf("%d%d",&n,&m);
    for(int i = 1; i <=  3 * n + 1; i++)
    dis[i] = -23333333;
    for(int i = 1; i <= n; i++)
    scanf("%d",&val[i]);
    add(n, 3 * n + 1, 0);
    add(3 * n, 3 * n + 1, 0);
    for(int i = 1; i <= m; i++)
    {
        int u, v, w;
        scanf("%d%d%d",&u,&v,&w);
        if(w == 1)
        {
            add(u, v, 0);
            add(u + n, v + n, 0);
            add(u, v + n, -val[u]);
            add(u + n * 2, v + n * 2, 0);
            add(u + n, v + n * 2, val[u]);
        }
        else
        {
            add(u, v, 0);
            add(u + n, v + n, 0);
            add(u, v + n, -val[u]);
            add(u + n * 2, v + n * 2, 0);
            add(u + n, v + n * 2, val[u]);
            add(v, u, 0);
            add(v + n, u + n, 0);
            add(v, u + n, -val[v]);
            add(v + n * 2, u + n * 2, 0);
            add(v + n, u + n * 2, val[v]);
        }
    }
    q.push(1);
    dis[1] = 0;
    vis[1] = 1;
    SPFA();
    printf("%d\n",dis[3 * n + 1]);
    return 0;
}

【luogu P1073 最優貿易】 題解