CSU 2172 回溯

http://acm.csu.edu.cn:20080/csuoj/problemset/problem?pid=2172

Description

ICPCCamp 有 n 个商店，用 1, 2, …, n 编号。对于任意 i > 1，有从商店 pi 到 i 的单向道路。 同时，商店 i 出售类型为 ai 的商品。

Bobo 从商店 1 出发前往商店 i。他要在两个不同的商店购买商品（包括商店 1 和 i）。设他先购买的商品类型是 x，后购买的商品类型是 y，他用 fi 表示不同的有序对 〈x, y〉 的数量。 求出 f2, f3, …, fn 的值。

• 1 ≤ n ≤ 105
• 1 ≤ pi < i
• 1 ≤ ai ≤ n
• n 的总和不超过 5 × 105.

Input

输入文件包含多组数据，请处理到文件结束。

每组数据的第一行包含 1 个整数 n.

第二行包含 (n − 1) 个整数 p2, p3, …, pn.

第三行包含 n 个整数 a1, a2, …, an.

Output

对于每组数据输出 (n − 1) 个整数表示 f2, f3, …, fn.

Sample Input

3
1 2
1 2 3
3
1 1
1 2 3
4
1 2 3
1 3 2 3

Sample Output

1
3
1
1
1
3
5

Hint

对于第三个样例，当 i = 4 时，可能的有序对 〈x, y〉 有 〈1, 2〉,〈1, 3〉,〈2, 3〉,〈3, 2〉,〈3, 3〉 共 5 种。所以 f4 = 5.

Source

2018湖南省第14届大学生计算机程序设计竞赛

Author

ftiasch

思路：设ans[i]表示从商店1到商店i的不同的有序对<x，y>的个数(即题目中的fi)，pre[i]表示上一次以i为第二分量的二元组的个数，a[i]表示商店i卖的商品的种类，假如我们现在在商店i，下一个要去的商店是j，且已知ans[i]、pre[a[j]]以及从根节点1到当前节点i的不同的商品种类数，那么有：ans[j]=ans[i]+cnt-pre[a[j]](去重)。这就是本道题的关键所在，而这些信息都可以在dfs中进行维护。注意该题会爆int。

#include <bits/stdc++.h>
#define INF 0x3f3f3f3f
using namespace std;
typedef long long ll;

int n;
vector<int> Edge[100005];
int a[100005];
ll pre[100005];//记录上一次以i作为第二分量的二元组的个数
int vis[100005];//记录商品类型
ll ans[100005];//记录答案

void dfs(int cur,int cnt)
{
	int len=Edge[cur].size();
	for(int i=0;i<len;i++)
	{
		int to=Edge[cur][i];//下一个商店
		if(!vis[a[cur]])//记录从根节点到当前节点不同的商品种类数
			++cnt;
		++vis[a[cur]];
		ans[to]=ans[cur]+cnt-pre[a[to]];
		ll v=pre[a[to]];
		pre[a[to]]=cnt;
		dfs(to,cnt);
		--vis[a[cur]];//回溯
		if(vis[a[cur]]==0)
			--cnt;
		pre[a[to]]=v;
	}
}

int main()
{
	while(~scanf("%d",&n))
	{
		for(int i=1;i<=n;i++)
			Edge[i].clear();
		memset(vis,0,sizeof(vis));
		memset(pre,0,sizeof(pre));
		int t;
		for(int i=2;i<=n;i++)
		{
			scanf("%d",&t);
			Edge[t].push_back(i);
		}
		for(int i=1;i<=n;i++)
			scanf("%d",&a[i]);
		dfs(1,0);
		for(int i=2;i<=n;i++)
			printf("%lld\n",ans[i]);
	}
	return 0;
}