商汤科技2018.9.7----软件开发笔试第三题

原创于 2018-09-07 21:41:18 发布 · 1.5k 阅读

2 ·

CC 4.0 BY-SA版权

文章标签：

#商汤科技笔试

笔试题专栏收录该内容

2 篇文章

订阅专栏

本文介绍了一种使用动态规划解决算法竞赛中特定问题的方法。通过维护额外的数据结构和数组，如next和save_A，实现了一个迭代过程来找到从任意点开始所能达到的最大得分。这个过程涉及查找集合中未访问过的节点，更新保存的得分，并检查是否达到了遍历的终点。

//除不变的A，B外加一个next数组，再加一个保存以i+1为起始点，一直走下去所得的分数
//，修改next数组，如果不再增加点，遍历结束，退出
#include<vector>
#include<iostream>
#include<sstream>
#include<string>
#include<set>
#include<algorithm>
#include<map>
using namespace std;


int main()
{
    int N;
    while (cin >> N)
    {
        vector<int> A, B;
        int mark_a, mark_b;
        for (int i = 0; i < N; ++i)
        {
            cin >> mark_a >> mark_b;
            A.push_back(mark_a);
            B.push_back(mark_b);
        }
        int max_index = 0;
        for (int i = 0; i < N; ++i)
        {
            if (A[i] > A[max_index])max_index = i;
        }
        int max_value = A[max_index];

        vector<set<int>> the_set(N);
        vector<int> next = B;
        vector<int> save_A = A;
        while (1)
        {
            bool flag = true;
            for (int i = 0; i < N; ++i)
            {
                if (find(the_set[i].begin(), the_set[i].end(), next[i])==the_set[i].end())
                {
                    flag = false;
                    save_A[i] += A[next[i]-1];
                    if (save_A[i] > max_value)
                    {
                        max_value = save_A[i];
                    }
                    the_set[i].insert(next[i]);
                }
                next[i] = B[next[i]-1];
            }
            if (flag == true)
            {
                break;
            }
        }
        cout << max_value << endl;

    }
    return 0;
}