题意:给一棵树n个点,m次操作,每次把一条x到y的链涂一个Z颜色
问最后每个节点,对应被涂过最多的颜色是什么?
先树剖一下,变成: m次操作,每次给一个区间染色,最后问每个点被染过最多次的颜色是什么
这个问题的离线方法真的很巧妙啊。。。
根据颜色开一个线段树,第i位置表示颜色i出现过东西次
开一个vector<int> s[maxn],
假设在【l,r】中涂上颜色c,则s[l].push(c),s[r+1].push(-c)
最后对每一个点i, 依次取出所有值, 如果s[i][j]==+c,则在线段树的位置c 加1,若==-c,则减一。
对于i的每个s[i][j],操作完取颜色线段树的最大值对应的颜色就是i点的答案
//BZOJ4034
#include <stdio.h>
#include <string.h>
#include <iostream>
#include <algorithm>
#include <vector>
#include <queue>
#include <set>
#include <map>
#include <string>
#include <math.h>
#include <stdlib.h>
#include <time.h>
#define lson l , m , rt << 1
#define rson m + 1 , r , rt << 1 | 1
using namespace std;
typedef long long ll;
const int MAXN = 100000+50;
vector<int > s[MAXN];
struct Edge
{
int to,next;
} edge[MAXN*2];
int head[MAXN],tot;
int top[MAXN];//top[v]表示v所在的重链的顶端节点
int fa[MAXN]; //父亲节点
int deep[MAXN];//深度
int num[MAXN];//num[v]表示以v为根的子树的节点数
int p[MAXN];//p[v]表示v与其父亲节点的连边在线段树中的位置
int fp[MAXN];//和p数组相反
int son[MAXN];//重儿子
int pos;
int out[MAXN];//dfs序
void init()
{
tot = 0;
memset(head,-1,sizeof(head));
pos = 0;
memset(son,-1,sizeof(son));
}
void addedge(int u,int v)
{
edge[tot].to = v;
edge[tot].next = head[u];
head[u] = tot++;
}
void dfs1(int u,int pre,int d) //第一遍dfs求出fa,deep,num,son
{
deep[u] = d;
fa[u] = pre;
num[u] = 1;
for(int i = head[u]; i != -1; i = edge[i].next)
{
int v = edge[i].to;
if(v != pre)
{
dfs1(v,u,d+1);
num[u] += num[v];
if(son[u] == -1 || num[v] > num[son[u]])
son[u] = v;
}
}
}
void getpos(int u,int sp) //第二遍dfs求出top和p
{
top[u] = sp;
p[u] = ++pos;
fp[p[u]] = u;
if(son[u]!=-1)
getpos(son[u],sp);
for(int i = head[u] ; i != -1; i = edge[i].next)
{
int v = edge[i].to;
if(v != son[u] && v != fa[u])
getpos(v,v);
}
out[u]=pos;
}
const int maxn = 100000+5;
struct TREE
{
int st_max[maxn<<2];
void build()
{
memset(st_max,0,sizeof st_max);
}
void PushUP(int rt)
{
st_max[rt] = max(st_max[rt<<1],st_max[rt<<1|1]);
}
void update(ll l,ll r,ll rt,ll num,ll val)
{
if (l == r&& l==num)
{
st_max[rt]+=val;
return ;
}
ll m = (l + r) >> 1;
if (num<=m)
update(lson,num,val);
else
update(rson,num,val);
PushUP(rt);
}
int query_max( int l,int r,int rt) //L,R为为查询区间,lr当前区间
{
if (l==r) return l;
int m = (l + r) >> 1;
if (st_max[rt<<1]>=st_max[rt<<1|1])
return query_max( lson);
else query_max( rson);
}
ll findsum(int u,int v,int z)//查询u->v链的sum
{
int f1=top[u],f2=top[v];
ll tmp=0;
while(f1!=f2)
{
if (deep[f1]<deep[f2])
{
swap(f1,f2);
swap(u,v);
}
s[p[f1]].push_back(+z);
s[p[u]+1].push_back(-z);
//tmp+=(tmp,query(1,1,pos,p[f1],p[u]));
u=fa[f1],f1=top[u];
}
if (deep[u]>deep[v] ) swap(u,v);
s[p[u]].push_back(+z);
s[p[v]+1].push_back(-z);
// return (tmp+ query(1,1,pos,p[u],p[v])); //若val(u)是u到fu的边权,则用son[u]
}
};
TREE tp;
int ans[MAXN];
int main()
{
int n,m;
while(scanf("%d%d",&n,&m)!=EOF)
{
if (!n&&!m) break;
init();
for (int i=1; i<=n+1; i++)
s[i].clear();
int u,v;
for (int i=1; i<n; i++)
{
scanf("%d%d",&u,&v);
addedge(u,v);
addedge(v,u);
}
dfs1(1,0,0);
getpos(1,1);
tp.build();
int op,z;
for (int i=1; i<=m; i++)
{
scanf("%d%d%d",&u,&v,&z);
tp.findsum(u,v,z);
}
}
return 0;
}