Description
给出一棵
n
个节点的树,根节点为
1.
+ v
,表示
v
住户有小偷进入,保证之前
2.
− v
表示
v
住户里的小偷离开,保证之前
每次操作结束后,问至少需要断几条树边可以使得所有有小偷的住户没有水喝,在断边数最少的前提下,使得没有进小偷但被断水的住户数最少
Input
第一行两个整数
n,q
分别表示点数和操作数,之后输入
n−1
个整数分别表示
2,3,...,n
节点的父亲节点,最后
q
行每行一个操作
Output
对于每个操作输出两个整数分别表示操作结束后使得所有有小偷的住户没有水和需要断的最少树边数,及最少的被断水但是没有进小偷的住户数
Sample Input
7 6
1 2 1 3 3 3
+
4
+
6
−
6
Sample Output
1 0
2 0
2 1
2 0
2 1
2 0
Solution
考虑所有
1
的儿子节点
Code
#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<vector>
#include<queue>
#include<map>
#include<set>
#include<ctime>
using namespace std;
typedef long long ll;
#define maxn 100005
int n,q;//fa[i][j]表示i的2^j级祖先
int fa[maxn][18];
int belong[maxn];//belong[i]表示i属于哪棵子树
int Size[maxn];//Size[i]表示以i为根的子树中叶子节点个数
int Num[maxn];//Num[i]表示以i为根的子树中被标记的点的个数
int Deep[maxn];//Deep[i]表示i节点的深度
int Index;
int dfn[maxn];//dfn[i]表示i节点的dfs序
int ID[maxn];//ID[i]表示以i为dfs序的节点编号
vector<int>g[maxn];
set<int>s[maxn];
void dfs(int u,int f,int dep)
{
Deep[u]=dep;
belong[u]=f;
dfn[u]=++Index;
ID[Index]=u;
if(!g[u].size())Size[u]=1;
for(int i=0;i<g[u].size();i++)
{
int v=g[u][i];
dfs(v,f,dep+1);
Size[u]+=Size[v];
}
}
int lca(int a,int b)
{
int i,j;
if(Deep[a]<Deep[b])swap(a,b);
for(i=0;(1<<i)<=Deep[a];i++);
i--;
for(j=i;j>=0;j--)
if(Deep[a]-(1<<j)>=Deep[b])
a=fa[a][j];
if(a==b) return a;
for(j=i;j>=0;j--)
{
if(fa[a][j]&&fa[a][j]!=fa[b][j])
{
a=fa[a][j];
b=fa[b][j];
}
}
return fa[a][0];
}
int main()
{
freopen("gangsters.in","r",stdin);
freopen("gangsters.out","w",stdout);
while(~scanf("%d%d",&n,&q))
{
for(int i=1;i<=n;i++)g[i].clear(),s[i].clear();
memset(fa,0,sizeof(fa));
memset(Size,0,sizeof(Size));
memset(Num,0,sizeof(Num));
for(int i=2;i<=n;i++)
{
int j;
scanf("%d",&j);
fa[i][0]=j,g[j].push_back(i);
}
Deep[1]=0;
Index=0;
for(int i=0;i<g[1].size();i++)
{
dfs(g[1][i],g[1][i],1);
Size[1]+=Size[g[1][i]];
}
for(int i=1;i<=n;i++)
for(int j=1;j<18;j++)
fa[i][j]=fa[fa[i][j-1]][j-1];
int ans1=0,ans2=0;
while(q--)
{
char op[3];
int v;
scanf("%s%d",op,&v);
int root=belong[v];
if(op[0]=='+')
{
if(!Num[root])
{
ans1++;
Num[root]++;
s[root].insert(dfn[v]);
}
else
{
ans2-=Size[lca(ID[*s[root].begin()],ID[*--s[root].end()])]-Num[root];
Num[root]++;
s[root].insert(dfn[v]);
ans2+=Size[lca(ID[*s[root].begin()],ID[*--s[root].end()])]-Num[root];
}
}
else
{
if(Num[root]==1)
{
ans1--;
Num[root]=0;
s[root].erase(dfn[v]);
}
else
{
ans2-=Size[lca(ID[*s[root].begin()],ID[*--s[root].end()])]-Num[root];
Num[root]--;
s[root].erase(dfn[v]);
ans2+=Size[lca(ID[*s[root].begin()],ID[*--s[root].end()])]-Num[root];
}
}
printf("%d %d\n",ans1,ans2);
}
}
return 0;
}