传送门


思路

官方题解是倒推,这里提供一种正推的做法。

不知道你们是怎么想到倒推的……感觉正推更好想啊QwQ就是不好码

把每一条红边,将其转化为蓝树上的一条路径。为了连这条红边,需要保证这条路径仍然完整。

考虑连完之后要删掉的那条蓝边,显然它只能被当前连的红路径覆盖而没有被其他路径覆盖,否则就玩不下去了。

即:每次只能删掉一条被覆盖一次的蓝边。

有没有可能一条蓝边没有被覆盖过呢?显然是不可能的,因为每条蓝边最终都要删去,若没有覆盖过则无法删去。

那么做法就很显然了:树剖,线段树维护最小值和覆盖了这个区间的红边编号。每次找一个只被覆盖了一次的蓝边,删掉它(将它的值赋为\(\infty\)),也去除那条红边的影响。如果能做到最后就是合法,否则不合法。

然而我太久没写树剖,而且\(n\)和\(n-1\)总分不清,结果贡献了一页的评测记录……

而且在毒瘤的洛谷上选c++11交到AtCoder上就变成了c++14……对AtCoder不熟,各种奇怪的错误都来了。

代码巨丑,凑合着看吧QwQ


代码

#include<bits/stdc++.h>
namespace my_std{
using namespace std;
#define pii pair<int,int>
#define fir first
#define sec second
#define MP make_pair
#define rep(i,x,y) for (int i=(x);i<=(y);i++)
#define drep(i,x,y) for (int i=(x);i>=(y);i--)
#define go(x) for (int i=head[x];i;i=edge[i].nxt)
#define sz 101010
typedef long long ll;
template<typename T>
inline void read(T& t)
{
t=0;char f=0,ch=getchar();
double d=0.1;
while(ch>'9'||ch<'0') f|=(ch=='-'),ch=getchar();
while(ch<='9'&&ch>='0') t=t*10+ch-48,ch=getchar();
if(ch=='.')
{
ch=getchar();
while(ch<='9'&&ch>='0') t+=d*(ch^48),d*=0.1,ch=getchar();
}
t=(f?-t:t);
}
template<typename T,typename... Args>
inline void read(T& t,Args&... args){read(t); read(args...);}
// void file()
// {
// #ifndef ONLINE_JUDGE
// freopen("a.txt","r",stdin);
// #endif
// }
// inline ll mul(ll a,ll b){ll d=(ll)(a*(double)b/mod+0.5);ll ret=a*b-d*mod;if (ret<0) ret+=mod;return ret;}
}
using namespace my_std;
// clock_t t=clock(); int n;
struct hh{int t,nxt;}edge[sz<<1];
int head[sz],ecnt;
void make_edge(int f,int t)
{
edge[++ecnt]=(hh){t,head[f]};
head[f]=ecnt;
edge[++ecnt]=(hh){f,head[t]};
head[t]=ecnt;
}
int size[sz],son[sz],top[sz],fa[sz],dep[sz],dfn[sz],cnt=-1;
#define v edge[i].t
void dfs1(int x,int fa)
{
dep[x]=dep[::fa[x]=fa]+1;size[x]=1;
go(x) if (v!=fa)
{
dfs1(v,x);
size[x]+=size[v];
if (size[v]>size[son[x]]) son[x]=v;
}
}
void dfs2(int x,int fa,int tp)
{
top[x]=tp;dfn[x]=++cnt;
if (son[x]) dfs2(son[x],x,tp);
go(x) if (v!=fa&&v!=son[x]) dfs2(v,x,v);
}
#undef v bool del[sz];
int F[sz],T[sz];
int mn[sz<<2],tag[sz<<2];
vector<int>tr[sz<<2];
#define ls k<<1
#define rs k<<1|1
#define lson ls,l,mid
#define rson rs,mid+1,r
void Add(int k,int t){tag[k]+=t;mn[k]+=t;}
void pushdown(int k){int &t=tag[k]; if (!t) return; Add(ls,t); Add(rs,t); t=0; return;}
void pushup(int k){mn[k]=min(mn[ls],mn[rs]);}
void add(int k,int l,int r,int x,int y,int t)
{
if (x<=l&&r<=y) return tr[k].push_back(t),Add(k,1);
pushdown(k);
int mid=(l+r)>>1;
if (x<=mid) add(lson,x,y,t);
if (y>mid) add(rson,x,y,t);
pushup(k);
}
void change(int k,int l,int r,int x,int y,int s)
{
if (x<=l&&r<=y) return Add(k,s);
pushdown(k);
int mid=(l+r)>>1;
if (x<=mid) change(lson,x,y,s);
if (y>mid) change(rson,x,y,s);
pushup(k);
}
int K;
int getmn(int k,int l,int r)
{
if (mn[k]>1) return -1; // -1 : don't have 1
int ret;
while (!tr[k].empty()&&del[tr[k].back()]) tr[k].pop_back();
ret=tr[k].empty()?0:tr[k].back(); // 0 : don't know which edge
if (l==r) return K=l,ret;
pushdown(k);
int mid=(l+r)>>1,cur;
cur=getmn(lson); if (cur!=-1) { if (cur) return cur; if (ret) return ret; }
cur=getmn(rson); if (cur!=-1) { if (cur) return cur; if (ret) return ret; }
return ret;
} void add(int x,int y,int t)
{
while (top[x]!=top[y])
{
if (dep[top[x]]<dep[top[y]]) swap(x,y);
add(1,1,n-1,dfn[top[x]],dfn[x],t);
x=fa[top[x]];
}
if (dfn[x]>dfn[y]) swap(x,y);
if (x!=y) add(1,1,n-1,dfn[x]+1,dfn[y],t);
}
void erase(int x,int y)
{
while (top[x]!=top[y])
{
if (dep[top[x]]<dep[top[y]]) swap(x,y);
change(1,1,n-1,dfn[top[x]],dfn[x],-1);
x=fa[top[x]];
}
if (dfn[x]>dfn[y]) swap(x,y);
if (x!=y) change(1,1,n-1,dfn[x]+1,dfn[y],-1);
}
void del_mn()
{
if (mn[1]!=1) puts("NO"),exit(0);
int e=getmn(1,1,n-1);
del[e]=1;
change(1,1,n-1,K,K,sz);
erase(F[e],T[e]);
} int main()
{
// file();
int x,y;
scanf("%d",&n);
rep(i,1,n-1) scanf("%d %d",&x,&y),make_edge(x,y);
dfs1(1,0);dfs2(1,0,1);
rep(i,1,n-1) scanf("%d %d",&x,&y),add(x,y,i),F[i]=x,T[i]=y;
rep(i,1,n-1) del_mn();
puts("YES");
// cout<<1.0*(clock()-t)/CLOCKS_PER_SEC;
return 0;
}
05-04 01:46