传送门
传送门

题意：给一棵树，树上有颜色，每种颜色有权值，定义一条路径的权值为所有颜色相同段的权值之和，求长度在$[L,R]$中的路径的最大权值。

数据范围：暴力过不了

显然是个点分治

对于分治中心考虑过中心的路径贡献的答案

（以下的“子树”指根直接与分治中心相连的子树）

把分治中心作为根，定义一棵子树的颜色为与根（即分治中心）相连的边的颜色

显然可以算出两边的贡献，如果两边颜色一样，再减掉这个权值

有个单调队列的神仙做法，看不懂

所以自己$yy$了线段树做法（题解也有）

把所有子树按颜色排序，把相同颜色放一起

维护两棵线段树，表示长度为$x$的路径的最大权值，一棵维护相同颜色，一棵维护不同颜色

遍历所有子树的所有节点，设当前节点的深度为$dis$，区间查询两棵线段树中$[L-dis,R-dis]$的答案。

遍历完一棵子树后，将它们加入同色线段树。

当进入不同颜色时，把前一个颜色的点插入异色线段树，并给同色线段树打清除标记

复杂度大概$O(Nlo{g}_N^2)$

#include <iostream>
#include <cstdio>
#include <cstring>
#include <cctype>
#include <algorithm>
#define MAXN 200005
#define MAXM 400005
#define re register
#define INF 0x3f3f3f3f
using namespace std;
class fast_input {
private:static const int SIZE = 1 << 15 | 1;char buf[SIZE], *front, *back;void Next(char &c) {if(front == back) back = (front = buf) + fread(buf, 1, SIZE, stdin);c = front == back ? (char)EOF : *front++;}public :template<class T>void operator () (T &x) {char c, f = 1;for(Next(c); !isdigit(c); Next(c)) if(c == '-') f = -1;for(x = 0; isdigit(c); Next(c)) x = x * 10 + c - '0';x *= f;}void operator () (char &c, char l = 'a', char r = 'z') {for(Next(c); c > r || c < l; Next(c)) ;}
}input;struct edge{int u,v,w;}e[MAXM];
int head[MAXN],nxt[MAXM],cnt;
void addnode(int u,int v,int w)
{e[++cnt]=(edge){u,v,w};nxt[cnt]=head[u];head[u]=cnt;
}
int cost[MAXN],n,m,L,R;
bool cut[MAXN];
int siz[MAXN],sum[MAXN],maxp[MAXN],root;
void findroot(int u,int f,int sum)
{siz[u]=1,maxp[u]=0;for (re int i=head[u];i;i=nxt[i])if (e[i].v!=f&&!cut[e[i].v]){findroot(e[i].v,u,sum);siz[u]+=siz[e[i].v];maxp[u]=max(maxp[u],siz[e[i].v]);}if (sum-siz[u]>maxp[u]) maxp[u]=sum-siz[u];if (maxp[u]<maxp[root]) root=u;
}
struct SGT
{#define lc p<<1#define rc p<<1|1int mx[MAXN<<2],cl[MAXN<<2];inline void pushcl(int p){mx[p]=-INF;cl[p]=1;}inline void pushdown(int p){if (cl[p]) pushcl(lc),pushcl(rc),cl[p]=0;}void build(int p,int l,int r){mx[p]=-INF;if (l==r) return;int mid=(l+r)>>1;build(lc,l,mid),build(rc,mid+1,r);}void modify(int p,int l,int r,int k,int v){if (l==r) return (void)(mx[p]=max(mx[p],v));pushdown(p);int mid=(l+r)>>1;if (k<=mid) modify(lc,l,mid,k,v);else modify(rc,mid+1,r,k,v);mx[p]=max(mx[lc],mx[rc]);}int query(int p,int l,int r,int ql,int qr){pushdown(p);if (ql<=l&&r<=qr) return mx[p];if (r<ql||qr<l) return -INF;int mid=(l+r)>>1;return max(query(lc,l,mid,ql,qr),query(rc,mid+1,r,ql,qr));}inline void clear(){pushcl(1);}
}sam,dif;
int ans=-INF,maxlen;
int nod[MAXN],col[MAXN];
inline bool cmp(const int& a,const int& b){return col[a]<col[b];}
int tot;
int len[MAXN],val[MAXN],num,_len[MAXN],_val[MAXN],_num;
int dfs(int u,int f,int l,int v,int las_col)
{int s=1;if (l>=L&&l<=R) ans=max(ans,v);maxlen=max(maxlen,l);len[++num]=l,val[num]=v;_len[++_num]=l,_val[_num]=v;for (int i=head[u];i;i=nxt[i])if (e[i].v!=f&&!cut[e[i].v])s+=dfs(e[i].v,u,l+1,e[i].w==las_col? v:v+cost[e[i].w],e[i].w);return s;
}
void calc()
{maxlen=0;for (re int i=head[root];i;i=nxt[i])if (!cut[e[i].v])nod[++tot]=e[i].v,col[e[i].v]=e[i].w;sort(nod+1,nod+tot+1,cmp);for (re int i=1;i<=tot;i++){if (col[nod[i]]!=col[nod[i-1]]){for (int u=1;u<=_num;u++) dif.modify(1,1,n,_len[u],_val[u]);sam.clear(),_num=0;}sum[nod[i]]=dfs(nod[i],0,1,cost[col[nod[i]]],col[nod[i]]);for (re int u=1;u<=num;u++){ans=max(ans,val[u]+sam.query(1,1,n,L-len[u],R-len[u])-cost[col[nod[i]]]);ans=max(ans,val[u]+dif.query(1,1,n,L-len[u],R-len[u]));}for (re int u=1;u<=num;u++) sam.modify(1,1,n,len[u],val[u]);num=0;}sam.clear(),dif.clear();tot=num=_num=0;
}
void solve()
{cut[root]=true;calc();for (re int i=head[root];i;i=nxt[i])if (!cut[e[i].v]){maxp[root=0]=INF;findroot(e[i].v,0,sum[e[i].v]);sum[e[i].v]=0,solve();}
}
int main()
{input(n),input(m),input(L),input(R);for (re int i=1;i<=m;i++) input(cost[i]);for (re int i=1;i<n;i++){int u,v,w;input(u),input(v),input(w);addnode(u,v,w),addnode(v,u,w);}sam.build(1,1,n),dif.build(1,1,n);maxp[root=0]=INF;findroot(1,0,n);solve();printf("%d\n",ans);return 0;
}