题意:给出一棵树,给出问题然后求解其最小共同祖先,以及出现的次数
输入格式:由于输入格式限定所以在scanf时要注意格式
样例格式
5
4:(0)
2:(1) 3
3:(0)
6
(1 5) (1 4) (4 2)
(2 3) (1 3) (4 3)
思路:这里用tarjan离线处理,由于tarjan是寻找find(x)的祖先,所以当find(x)被找到时用ans[]数组存储的祖先出现的次数加一
与Nearest Common Ancestors那道题相比较,这里使用isroot来判定根(上次使用indegree[]:逻辑一样即入度为零作为根)
然后对于图的存储仍使用的vector而没有使用链式向前星的做法.(感觉那个模板有点高级)
#include<iostream>
#include<cstring>
#include<algorithm>
#include<vector>
using namespace std;
const int maxn = 1000;
int que[maxn][maxn];//保存询问
int ans[maxn];//保存祖先i出现过的次数
int n,m;
vector<int> g[maxn];//保存儿子
int root;//树的根
bool visit[maxn];
bool isroot[maxn];//利用不为任何人的子树为根的性质标记根节点
int pre[maxn];
int find(int x)
{
if(pre[x] == x) return x;
else return pre[x] = find(pre[x]);
}
void init()
{
for(int i = 1; i <= n; i++)
{
ans[i] = 0;
visit[i] = false;
isroot[i] = true;
que[i] = 0;
g[i].clear();
pre[i] = i;
}
}
void tarjan(int root)
{
for(int i = 1; i <= n; i++)
{
if(visit[i]&&que[root][i])
{
ans[find(i)] += que[root][i];
}
}
visit[root] = true;
for(int i = 0; i < g[root].size(); i++)
{
int term = g[root][i];
tarjan(term);
pre[term] = root;
}
}
int main()
{
while(~scanf("%d",&n))
{
init();
int s,num;
for(int i = 1; i <= n; i++)
{
scanf("%d:(%d)",&f,&num);
for(int j = 1; j <= num; j++)
{
scanf(" %d",&s);
isroot[s] = false;
g[f].push_back(s);
}
}
//确定根
for(int i = 1; i <= n; i++)
{
if(isroot[i])
{
root = i;
break;
}
}
scanf("%d",&m);
int u,v;
for(int i = 1; i <= m; i++)
{
scanf(" (%d %d)",&u,&v);
que[u][v]++;
que[v][u]++;
}
tarjan(root);
for(int i = 1; i <= n; i++)
{
if(ans[i])
{
printf("%d:%d\n",i,ans[i]);
}
}
}
return 0;
}