P6086 【模板】Prufer 序列
题意
\(prufer\)序列和树的互相转化。
分析
模板题,要背一遍。
一些性质:
代码
#include<bits/stdc++.h>
using namespace std;
template <typename T>
inline void read(T &x){
x=0;char ch=getchar();bool f=false;
while(!isdigit(ch)){if(ch=='-'){f=true;}ch=getchar();}
while(isdigit(ch)){x=x*10+(ch^48);ch=getchar();}
x=f?-x:x;
return ;
}
template <typename T>
inline void write(T x){
if(x<0) putchar('-'),x=-x;
if(x>9) write(x/10);
putchar(x%10^48);
return ;
}
#define ll long long
const int N=5e6+5;
int n,m;
int fa[N],pru[N],d[N];
ll Ans;
void TreetoPrufer(){
for(int i=1;i<=n-1;i++) d[fa[i]]++;
for(int i=1,now=1;i<=n-1;i++,now++){
while(d[now]) now++;
pru[i]=fa[now];
while(i<n-2&&!--d[pru[i]]&&pru[i]<now) pru[i+1]=fa[pru[i]],i++;
}
return;
}
void PrufertoTree(){
for(int i=1;i<=n-2;i++) d[pru[i]]++;
pru[n-1]=n;
for(int i=1,now=1;i<=n-1;i++,now++){
while(d[now]) now++;
fa[now]=pru[i];
while(i<n-1&&!--d[pru[i]]&&pru[i]<now) fa[pru[i]]=pru[i+1],i++;
}
return ;
}
signed main(){
read(n),read(m);
if(m==1){
for(int i=1;i<=n-1;i++) read(fa[i]);
TreetoPrufer();
for(int i=1;i<=n-2;i++) Ans^=1ll*i*pru[i];
}
else{
for(int i=1;i<=n-2;i++) read(pru[i]);
PrufertoTree();
for(int i=1;i<=n-1;i++) Ans^=1ll*i*fa[i];
}
write(Ans);
return 0;
}