Test for Job(poj3249 拓撲+動態規劃)

題目鏈接


題意:從任意一個源點到目標點, 有向無環, 求路徑的最大權值和,權值有負數。

思路: 我的思路複雜了一點,用了兩個vector分別保存岀度和入度, 當縮到當前這個點,它的所有前驅的值已經確定,所以選擇最大的值疊加到當前點。動歸方程:

d[i] = max(d[i], d[vec[i][j]]+v[i]);//d[i]是i點的最大權值和, vec[i][j]是i的某一個前驅,選擇是否疊加這個點。

注意一點:權值有負, 所以ans初始值爲負無窮。

#include <iostream>
#include <cstring>
#include <cstdio>
#include <algorithm>
#include <string>
#include <cmath>
#include <queue>
#include <set>
#include <stack>
#include <fstream>
#include <vector>
using namespace std;
int v[100010];
int in[100010];
int d[100010];
vector <int> vec[100010];
vector <int> tmp[100010];
int main()
{
    int n, m;
    int s, e;
    while(scanf("%d%d", &n, &m) == 2)
    {
        queue<int> que;
        memset(in, 0, sizeof(in));
        int ans = -0x7ffffff;
        for(int i = 1; i <= n; i++)
        {
             vec[i].clear();
             tmp[i].clear();
             scanf("%d", &v[i]);
        }

        for(int i = 0; i < m; i++)
        {
            scanf("%d%d", &s, &e);
            vec[e].push_back(s);
            tmp[s].push_back(e);
            in[e]++;
        }
        for(int i = 1; i <= n; i++)
        {
            if(in[i] == 0)
            {
               d[i] = v[i];
                que.push(i);
            }
            else
                d[i] = -0xfffffff;
        }
        while(!que.empty())
        {
            int i = que.front(); que.pop();
            for(int j = 0; j < vec[i].size(); j++)
            {
                d[i] = max(d[i], d[vec[i][j]]+v[i]);
            }
            for(int j = 0; j < tmp[i].size(); j++)
            {
                in[tmp[i][j]]--;
                if(in[tmp[i][j]] == 0)
                {
                    que.push(tmp[i][j]);
                }
            }

        }

        for(int i = 1; i <= n; i++)
        {
            if(tmp[i].size() == 0)
            {
                ans = max(ans, d[i]);
            }
        }
        cout << ans << endl;
    }
    return 0;
}



發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章