AtCoder Grand Contest 018 F - Two Trees 欧拉回路+构造

题意

给你两棵树，每棵树的点编号为1-n，然后对于相同编号的点附上一个相同的值，使得两棵树的每个点的子树的值的和的绝对值为1

N<=100000

分析

这是一道脑洞题
首先第一个，每个点子树都有奇偶性，然后如果对于编号相同的点如果子树内奇偶性不一样，就是Impossible
如果都相同，考虑怎么构造，满足下面的原则：

子树内每个孩子1和-1相匹配

我们可以这样,把两棵树并成一个大的树，A树的点变成1-n，B树的点变成n+1-2n
如果两个相同编号的子树和是奇数，那么把这两个在对应新树上连一条边
两棵树的边在新树上保留
新建一个根把两个根连上去
这样的话每个点的度数都是偶数，然后你可以跑一遍欧拉回路
这样之后如果有一条u->v的边，其中u是原来在A树上的，v是原来在B树上的，那么ans[u]=1,否则就是-1

这个是为什么呢？可以这样想，奇数的点肯定是成对出现的，一条树边是-1，一条跨越的边+1，这样子树内的点（不包括当前点）肯定是一条从A去B 然后又从B回来A，子树内部的出=子树内部的入

好像还有一种构造的方法，我没有看，可以去看官方题解

代码

#include <bits/stdc++.h>
using namespace std;
const int N = 1000010;
inline int read()
{
  int p=0; int f=1; char ch=getchar();
  while(ch<'0' || ch>'9'){if(ch=='-') f=-1; ch=getchar();}
  while(ch>='0' && ch<='9'){p=p*10+ch-'0'; ch=getchar();}
  return p*f;
}

struct node{int x,y,next;}edge[N]; int len,first[N];
void ins(int x,int y)
{
  len++; edge[len].x=x; edge[len].y=y; edge[len].next=first[x]; first[x]=len;
  len++; edge[len].x=y; edge[len].y=x; edge[len].next=first[y]; first[y]=len;
}

int n,cnta[N],cntb[N]; int ans[N];

int s[N],top; bool vis[N];

void dfs(int x)
{
  while(1)
  {
    int k = first[x]; if(k==-1) break; first[x] = edge[k].next;
    int y = edge[k].y;
    if(!vis[k>>1])
    {
      vis[k>>1] = 1;
      dfs(y); s[++top] = k;
    }
  }
}

int main()
{

  len=1; memset(first,-1,sizeof(first)); n=read();
  for(int i=1;i<=n;i++){int x=read(); if(x==-1) x=0,ins(x,i); else{ins(x,i); cnta[x]++;}}
  for(int i=1;i<=n;i++){int x=read(); if(x==-1) x=0,ins(x,i+n); else{ins(x+n,i+n); cntb[x]++;}}
  for(int i=1;i<=n;i++)
  {
    ans[i] = -1; if((cnta[i]&1) != (cntb[i]&1)){printf("IMPOSSIBLE\n"); return 0;}
    if((cnta[i]&1) == 1 && (cntb[i]&1) == 1) ans[i] = 0;
    else ins(i,i+n);
  }

  printf("POSSIBLE\n");
  top=0; dfs(1);

  for(int i=top;i>=1;i--)
  {
    int k = s[i];
    if(edge[k].x >= 1 && edge[k].x <= n && edge[k].y <=2*n && edge[k].y >= n+1) ans[edge[k].x] = 1;
  }

  for(int i=1;i<=n;i++) printf("%d%c",ans[i]," \n"[i==n]);

  return 0;
}