P6753 思路纯暴力,但是暴力写一半也不会写了,求教壶关
#include<bits/stdc++.h>
using namespace std;
const int N=2e5+10;
struct Node{
int next,to;
}a[N];
int dfn[N],f[N][21],head[N],tot,n,q,root,w[N],ll[N],vv[N],deep[N];
set<int> ss;
void add(int u,int v){
a[++tot].to=v;
a[tot].next=head[u];
head[u]=tot;
}
void dfs(int u,int fa){
deep[u]=deep[fa]+1;
for(int i=head[u];i;i=a[i].next){
int v=a[i].to;
if(v==fa)continue ;
dfs(v,u);
dfn[u]=min(dfn[u],v);
}
}
void dfs1(int now,int num,int la){//往下走的顺序
if(num==0){
return ;
}
//cout<<now<<" "<<dfn[now]<<" ";
while(w[dfn[now]]==0&&dfn[now]!=now&&w[now]!=0){
w[dfn[now]]++;
w[now]--;
dfs1(dfn[now],num--,now);
}
if(w[dfn[now]]!=0&&w[la]!=0){
// dfs1(dfn[now],num);
// puts("SOV");
w[now]++;
w[la]--;
num=num-1;
}
}
void dfs2(int now,int la){
//cout<<now<<" "<<dfn[now]<<" ";
while(w[dfn[now]]==0&&dfn[now]!=now&&w[now]!=0){
w[dfn[now]]++;
w[now]--;
dfs2(dfn[now],now);
}
if(w[dfn[now]]!=0&&w[la]!=0){
// dfs1(dfn[now],num);
// puts("SOV");
w[now]++;
w[la]--;
// num=num-1;
}
}
int main(){
memset(dfn,0x3f,sizeof(dfn));
cin>>n>>q;
for(int i=1;i<=n;i++){
int x;
cin>>x;
if(x==0){
root=i;
}
else {
add(x,i);
}
}
dfs(root,0);
for(int i=1;i<=n;i++){
vv[i]=1;
}
for(int i=1;i<=n;i++){
if(dfn[i]==0x3f3f3f3f)dfn[i]=i;
ss.insert(dfn[i]);
// cout<<dfn[i]<<" ";
}
for(const auto& i:ss){
vv[i]--;
}
while(q--){
int op,num,qq=0;
cin>>op>>num;
if(op==1){
w[root]=num;
dfs1(root,num,0);
// for(int i=head[root];i;i=a[i].next){
// int v=a[i].to;
// if(w[v]!=0){
// // cout<<v<<endl;
// break;
// }
// }
}else {
for(int i=1;i<=n;i++)ll[i]=w[i];
w[num]=0;
dfs2(root,0);
int cnt=0;
for(int i=1;i<=n;i++)if(ll[i]!=w[i])cnt++;
//cout<<cnt<<endl;
}
for(int i=1;i<=n;i++)qq+=w[i];
// cout<<endl;
// cout<<qq<<" ";
for(int i=n;i>=1;i--){
if(w[i]!=0){
dfs1(i,qq,0);
}
}
for(int i=1;i<=n;i++){
cout<<w[i]<<" ";
}
cout<<endl;
}
return 0;
}