题目详情 - 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的操作。
解题过程:
- 用path保存路径,ans保存答案,用一个graph保存输入样例(我用了一个-1作为终结标志),定义一个 t 数组查找没有出现的节点。
- 在输入的时候可以直接使用push_back (开始我不知道,有点浪费空间),输入的过程中需要把每一个数字做个标记(t[node] = 1)。 然后需要把输入的每一行从到到大sort一遍(为了输出最小序列,注意我在最后加了-1)
- 遍历所有节点,找到根节点进入(如果t[i]==0 那么他就没在子节点出现过,就是根节点,只用查找一次)如果没想到这点,那么测试点5就过不去,其他都可以过。也相当于一个剪枝,把所有以子节点开头的树剪去了。
- 最后就是一个常规的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;
}