【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;
}