L2-026 小字辈--不要频繁的shink_to_fit的段错误血与泪警告

2020年3月21日 95次阅读

本题给定一个庞大家族的家谱，要请你给出最小一辈的名单。

输入格式：

输入在第一行给出家族人口总数 N（不超过 100 000 的正整数） —— 简单起见，我们把家族成员从 1 到 N 编号。随后第二行给出 N 个编号，其中第 i 个编号对应第 i 位成员的父/母。家谱中辈分最高的老祖宗对应的父/母编号为 -1。一行中的数字间以空格分隔。

输出格式：

首先输出最小的辈分（老祖宗的辈分为 1，以下逐级递增）。然后在第二行按递增顺序输出辈分最小的成员的编号。编号间以一个空格分隔，行首尾不得有多余空格。

输入样例：

9
2 6 5 5 -1 5 6 4 7

输出样例：

4
1 9

题目链接：https://pintia.cn/problem-sets/994805046380707840/problems/994805055679479808

 

这题的思路很简单，就是把关系映射图反着存，然后从根遍历下去就可以了，遍历到叶子节点时判断深度，如果深度较大就进行替换。然后记录一个坑点，之前的时候我学习STL容器时是被告知clear()是不会自动释放内存的，一般的做法是用临时变量swap交换等回收（当然今天在羣里讨论时有人说是只有vector这样用），也被卡过，后面就一直用临时变量swap的这种做法（专业一点就叫shink_to_fit），然后今天就被这道题卡了，原因可能就是dfs层数太深临时变量还没回收然后导致的段错误，记录一下这血和泪的教训，下面附上AC代码

#include<iostream>
#include<set>
#include<vector>
using namespace std;
set<int> s;
int ans=0;
vector<int> v[100005];
void dfs(int now,int de)
{
	if(v[now].size()==0)
	{
		if(de>ans)
		{
			ans=de;
//			set<int> tmp;//dfs层数太深没来得及回收会段错误 
//			tmp.swap(s);	
			s.clear();		
			s.insert(now);
		}
		else if(de==ans)
		{
			s.insert(now);
		}
		return;
	}
	for(int i=0;i<v[now].size();i++)
	{
		dfs(v[now][i],de+1);
	}
}
int main()
{
	int n,start;
	scanf("%d",&n);
	for(int i=1;i<=n;i++)
	{
		int tmp;
		scanf("%d",&tmp);
		if(tmp==-1)
			start=i;
		else
			v[tmp].push_back(i);
	}
	dfs(start,1);
	printf("%d\n",ans);
	for(set<int>::iterator it=s.begin();it!=s.end();it++)
	{
		if(it!=s.begin())
			printf(" ");
		printf("%d",*it);
	}
	return 0;
}

 

点赞