【BZOJ】3319: 黑白树(并查集+特殊的技巧/-树链剖分+线段树)

时间:2023-11-10 22:09:20

http://www.lydsy.com/JudgeOnline/problem.php?id=3319

以为是模板题就复习了下hld。。。。。。。。。。。。。。。。。。。。。。。。。。。。。

然后nlg^2n被tle成翔了。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。

然后看题解QAQ,,,这。。。

神题做法。。。待会再写。。。(upd:【BZOJ】3319: 黑白树

tle的hld:

#include <cstdio>
#include <cstring>
#include <cmath>
#include <string>
#include <iostream>
#include <algorithm>
#include <queue>
#include <set>
#include <map>
using namespace std;
typedef long long ll;
#define rep(i, n) for(int i=0; i<(n); ++i)
#define for1(i,a,n) for(int i=(a);i<=(n);++i)
#define for2(i,a,n) for(int i=(a);i<(n);++i)
#define for3(i,a,n) for(int i=(a);i>=(n);--i)
#define for4(i,a,n) for(int i=(a);i>(n);--i)
#define CC(i,a) memset(i,a,sizeof(i))
#define read(a) a=getint()
#define print(a) printf("%d", a)
#define dbg(x) cout << (#x) << " = " << (x) << endl
#define error(x) (!(x)?puts("error"):0)
#define rdm(x, i) for(int i=ihead[x]; i; i=e[i].next)
inline const int getint() { int r=0, k=1; char c=getchar(); for(; c<'0'||c>'9'; c=getchar()) if(c=='-') k=-1; for(; c>='0'&&c<='9'; c=getchar()) r=r*10+c-'0'; return k*r; } const int N=1e6+10;
int ihead[N], cnt;
struct dat { int next, to, id; }e[N<<1];
void add(int u, int v, int id) {
e[++cnt].next=ihead[u]; ihead[u]=cnt; e[cnt].to=v; e[cnt].id=id;
e[++cnt].next=ihead[v]; ihead[v]=cnt; e[cnt].to=u; e[cnt].id=id;
} int n, son[N], top[N], fa[N], sz[N], root, id[N], tot, ed[N], dep[N];
void dfs1(int x, int f) {
sz[x]=1; int y; fa[x]=f; dep[x]=dep[f]+1;
rdm(x, i) if(f!=(y=e[i].to)) {
dfs1(y, x);
sz[x]+=sz[y];
if(sz[y]>sz[son[x]]) son[x]=y;
}
}
void dfs2(int x, int tp) {
top[x]=tp; id[x]=++tot;
if(son[x]) dfs2(son[x], tp);
rdm(x, i) if(e[i].to!=fa[x] && e[i].to!=son[x]) dfs2(e[i].to, e[i].to);
}
void geted(int x, int t) {
ed[id[x]]=t;
rdm(x, i) if(e[i].to!=fa[x]) geted(e[i].to, e[i].id);
} #define lc x<<1
#define rc x<<1|1
#define lson l, mid, lc
#define rson mid+1, r, rc
#define MID (l+r)>>1
struct T { int mx; bool tag; }t[N<<2];
void upd(int x, int r) {
t[x].tag=1;
t[x].mx=max(t[x].mx, r);
}
void pushdown(int x, int mid, int r) {
if(t[x].tag) t[x].tag=0, upd(lc, mid), upd(rc, r);
}
void pushup(int x) { t[x].mx=max(t[lc].mx, t[rc].mx); }
void update(int L, int R, int l=1, int r=tot, int x=1) {
if(L<=l && r<=R) { upd(x, r); return; }
int mid=MID;
pushdown(x, mid, r);
if(L<=mid) update(L, R, lson);
if(mid<R) update(L, R, rson);
pushup(x);
}
int ask(int L, int R, int l=1, int r=tot, int x=1) {
if(L<=l && r<=R) return t[x].mx;
int mid=MID, ret=0;
pushdown(x, mid, r);
if(L<=mid) ret=max(ret, ask(L, R, lson));
if(mid<R) ret=max(ret, ask(L, R, rson));
return ret;
}
void ask(int x) {
int fx=top[x], pos=0;
while(fx!=root) {
pos=ask(id[fx], id[x]);
if(pos!=0) break;
x=fa[fx]; fx=top[x];
}
if(pos==0 && x!=root) pos=ask(id[top[x]], id[x]);
printf("%d\n", ed[pos]);
}
void change(int x, int y) {
int fx=top[x], fy=top[y];
while(fx!=fy) {
if(dep[fx]<dep[fy]) { swap(fx, fy); swap(x, y); }
update(id[fx], id[x]);
x=fa[fx]; fx=top[x];
}
if(x==y) return;
if(dep[x]<dep[y]) swap(x, y);
update(id[y]+1, id[x]);
} int main() {
read(n); int m=getint();
for1(i, 1, n-1) { int u=getint(), v=getint(); add(u, v, i); }
root=1;
dfs1(root, 0); dfs2(root, root); geted(root, 0);
while(m--) {
int cc=getint();
if(cc==1) ask(getint());
else { int u=getint(), v=getint(); change(u, v); }
}
return 0;
}

  


Description

给定一棵树,边的颜色为黑或白,初始时全部为白色。维护两个操作:
 
 
1.查询u到根路径上的第一条黑色边的标号。
2.将u到v    路径上的所有边的颜色设为黑色。
 
Notice:这棵树的根节点为1

Input

第一行两个数n,m分别表示点数和操作数。
接下来n-?    1行,每行2个数u,v.表示一条u到v的边。
接下来m行,每行为以下格式:
 
 
1 v 表示第一个操作
 
 
2 v u 表示第二种操作

Output

对于每个询问,输出相应答案。如果不存在,输出0。

Sample Input

5 4
1 2
1 3
2 4
2 5
1 2
2 2 3
1 3
1 4

Sample Output

0
2
1

HINT

对于    100%    的数据:n,m<=10^6

Source