吐槽
这道题调了7个小时也是够了。最后只好比着题解做了一遍2333
思路
首先考虑n=2000的情况。因为这是在一条路径上，所以可以考虑差分。用a[i][j]表示第i个点中j这种粮食出现的次数。加入要在从x到y的路径上加入c这种粮食。将这条路径分为两部分进行差分。从x到lca，也就是将a[x][c]++,a[fa[lca]]--。从y到son[lca]。就是将a[y][c]++,a[lca]--。(详细原因见树上差分)最后dfs一遍，并且在回溯的时候合并一下就行了。
那么对于n=100000的数据，只要用线段树合并优化一下就行了。也就是将a数组改为n棵动态开点线段树。合并的时候用线段树合并的板子合并一下就行了。
代码
/*
* @Author: wxyww
* @Date: 2018-12-10 14:37:11
* @Last Modified time: 2018-12-10 21:15:06
*/
#include<cstdio>
#include<iostream>
#include<vector>
using namespace std;
typedef long long ll;
#define Ls TR[cur].ls
#define Rs TR[cur].rs
ll read() {
ll x = 0,f = 1;char c = getchar();
while(c < '0' || c > '9') {
if(c == '-') f = -1;
c = getchar();
}
while(c >= '0' && c <= '9') {
x = x * 10 + c - '0';
c = getchar();
}
return x * f;
}
const int N = 100000 + 100,logN = 20;
struct node {
int ls,rs,ans,sum;
}TR[N * 50];
int n,m,dep[N],lca[N][logN + 30],root[N];
vector<int>e[N];
int num;
void get_lca(int u,int father) {
for(int i = 1;i < logN;++i) lca[u][i] = lca[lca[u][i - 1]][i - 1];
int k = e[u].size();
for(int i = 0;i < k;++i) {
int v = e[u][i];
if(v == father) continue;
lca[v][0] = u;
dep[v] = dep[u] + 1;
get_lca(v,u);
}
}
int LCA(int x,int y) {
if(dep[x] < dep[y]) swap(x,y);
for(int i = logN - 1;i >= 0;--i)
if(dep[lca[x][i]] >= dep[y]) x = lca[x][i];
if(x == y) return y;
for(int i = logN - 1;i >= 0;--i) {
if(lca[x][i] != lca[y][i]) {
x = lca[x][i];y = lca[y][i];
}
}
return lca[x][0];
}
void up(int cur) {
if(TR[Ls].sum >= TR[Rs].sum) {
TR[cur].sum = TR[Ls].sum;
TR[cur].ans = TR[Ls].ans;
}
else {
TR[cur].sum = TR[Rs].sum;
TR[cur].ans = TR[Rs].ans;
}
}
void update(int &cur,int l,int r,int pos,int c) {
if(!cur) cur = ++num;
if(l == r) {
TR[cur].sum += c;
TR[cur].ans = l;
return;
}
int mid = (l + r) >> 1;
if(pos <= mid) update(Ls,l,mid,pos,c);
else update(Rs,mid + 1,r,pos,c);
up(cur);
}
int merge(int cur,int a,int l,int r) {
if(!cur) return a;
if(!a) return cur;
if(l == r) {
TR[cur].sum += TR[a].sum;
TR[cur].ans = l;
return cur;
}
int mid = (l + r) >> 1;
Ls = merge(Ls,TR[a].ls,l,mid);
Rs = merge(Rs,TR[a].rs,mid + 1,r);
up(cur);
return cur;
}
void dfs(int u,int father) {
int k = e[u].size();
for(int i = 0;i < k;++i) {
int v = e[u][i];
if(v == father) continue;
dfs(v,u);
merge(root[u],root[v],1,N - 100);
}
}
int main() {
int n = read(),m = read();
for(int i = 1;i < n;++i) {
int x = read(),y = read();
e[x].push_back(y);
e[y].push_back(x);
}
dep[1] = 1;
for(int i = 1;i <= n;++i) root[i] = ++num;
get_lca(1,0);
while(m--) {
int x = read(),y = read(),c = read();
int K = LCA(x,y);
update(root[x],1,N - 100,c,1);
update(root[y],1,N - 100,c,1);
update(root[K],1,N - 100,c,-1);
if(lca[K][0]) update(root[lca[K][0]],1,N - 100,c,-1);
}
dfs(1,0);
for(int i = 1;i <= n;++i) {
if(TR[root[i]].sum == 0) puts("0");
else
printf("%d
",TR[root[i]].ans);
}
return 0;
}