【PTA】L2-2 病毒溯源 回溯算法

题目详情 - L2-2 病毒溯源 (pintia.cn) //题目网址

输入样例:

10
3 6 4 8
0
0
0
2 5 9
0
1 7
1 2
0
2 3 1

输出样例:

4
0 4 9 1

大致思路:

刚开始分析题目输入输出样例时 会发现它是一种类似树的结构一层一层的去找。在寻找途中记录过程(用path保存)然后不断比较长度, 找到长度更长的就赋值给ans。

!!!我们需要从根节点进入这个树,因为只有带有根节点的树才是最长的!!!

整个过程就是一个dfs的操作。

解题过程:

  1. 用path保存路径,ans保存答案,用一个graph保存输入样例(我用了一个-1作为终结标志),定义一个 t 数组查找没有出现的节点。
  2. 在输入的时候可以直接使用push_back (开始我不知道,有点浪费空间),输入的过程中需要把每一个数字做个标记(t[node] = 1)。 然后需要把输入的每一行从到到大sort一遍(为了输出最小序列,注意我在最后加了-1
  3. 遍历所有节点,找到根节点进入(如果t[i]==0 那么他就没在子节点出现过,就是根节点,只用查找一次)如果没想到这点,那么测试点5就过不去,其他都可以过。也相当于一个剪枝,把所有以子节点开头的树剪去了。
  4. 最后就是一个常规的dfs没什么好说的。只是我用了一个while。当然用for也可以(我刚开始思路不对的时候用的,就懒得改回来了)

我感觉我写的蛮清晰的(dfs还可以写的再简单一点,但这样方便理解一点),容器的空间也没有浪费。但也浪费了很多时间。主要是思路对,但细节不对。考试的时候还是得多多看题,在脑子里面大致过一遍。

#include <bits/stdc++.h>
using namespace std;

int n, m;
vector<int> ans;
vector<int> path;
int t[10010];

void dfs(vector<vector<int>> &graph, int start, int length)
{
    int j = 0;
    if (graph[start][j] == -1)//
    {
        if (path.size() > ans.size())
        {
            ans.clear();
            ans = path;
        }
        return;
    }
    while (graph[start][j] != -1)
    {
        path.push_back(graph[start][j]);
        dfs(graph, graph[start][j], length + 1);
        path.pop_back();
        j++;
    }
}

int main()
{
    cin >> n;

    vector<vector<int>> graph(n, vector<int>());
    for (int i = 0; i < n; i++)
    {
        int j = 0;
        cin >> m;
        for (; j < m; j++)
        {
            int node;
            cin >> node;
            t[node] = 1;
            graph[i].push_back(node);
        }
        graph[i].push_back(-1);
        sort(graph[i].begin(), graph[i].begin() + j);
    }

    for (int i = 0; i < n; i++)
    {
        if (!t[i])
        {
            path.clear();
            path.push_back(i);
            dfs(graph, i, 1);
            break;
        }
    }

    cout << ans.size() << endl;
    for (auto it = ans.begin(); it != ans.end(); it++)//输出答案
    {
        if (it != ans.begin())
            cout << " ";
        cout << *it;
    }
    return 0;
}

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值