PAT（甲级） 1004 （树的遍历简易DFS））

额额，其实是颗假树，（笑
唉，还是读题太死了，虽然通常情况下01 02 这类输入需要用字符串作为输入，但。。。这题直接当int搞就行了！！！ 测评机输入的是“真整数”。不会影响的输入（服

我学到了：再一次领悟到了map vector 万能嵌套的魅力~

没搞懂的地方是：
还有，VS老是报错vector越界？？？ 我开100，用一个ans[0] （VS: 不行不行，你越界）
-----------------------------------------------------分割线---------------------------------------------------------------------
最后还是 弄明白了：原来是。 vector.clear() 不用能来清零！！
vector.clear()的真正作用是：把size设置成0，capacity不变。
所以我的vector ans 在清零后一个都不剩。
正确的清零用法是

1· 重新申请并分配：
ans.assign(100,0);

//assign先删除掉ans中的所有元素
2· 或者直接赋值0
for(int i = 0; i < (int)vector.size(); i++) vector[i] = 0;

即可清零！

#include<iostream>
#include<string>
#include<map>
#include<vector>
#include<algorithm>
using namespace std;
int maxd;
map<string, vector<string> > mp;
vector <int> ans(100);
void DFS(string cur, int d)
{
	maxd = max(maxd, d);
	if (mp[cur].empty()) {
		ans[d]++;
		return;
	}
	for (int i = 0; i < (int)mp[cur].size(); i++)
	{
		DFS(mp[cur][i],d+1);
	}
}

int main(void)
{
	freopen("in.txt", "r", stdin);
	int n, m;
	while (cin >> n && n)
	{
		cin >> m;
		mp.clear();
		//ans.clear(); //错误点
		ans.assign(100,0);
		while (m--)
		{
			string s;
			cin >> s;
			int k;
			cin >> k;
			while (k--)
			{
				string son;
				cin >> son;
				mp[s].push_back(son);
			}
		}
		maxd = 0;  //最大深度
		DFS("01", 0);
		for (int i = 0; i <= maxd; i++)
			cout << ans[i] << (!(maxd - i) ? "\n" : " ");

	}
}