# [BZOJ4012] [HNOI2015]开店

## Sample Input

10 10 10
0 0 7 2 1 4 7 7 7 9
1 2 270
2 3 217
1 4 326
2 5 361
4 6 116
3 7 38
1 8 800
6 9 210
7 10 278
8 9 8
2 8 0
9 3 1
8 0 8
4 2 7
9 7 3
4 7 0
2 2 7
3 2 1
2 3 4


## Sample Output

1603
957
7161
9466
3232
5223
1879
1669
1282
0


## Solution

#include<bits/stdc++.h>
using namespace std;

#define ll long long

x=0;int f=1;char ch=getchar();
for(;!isdigit(ch);ch=getchar()) if(ch=='-') f=-f;
for(;isdigit(ch);ch=getchar()) x=x*10+ch-'0';x*=f;
}

void print(ll x) {
if(x<0) putchar('-'),x=-x;
if(!x) return ;print(x/10),putchar(x%10+48);
}
void write(ll x) {if(!x) putchar('0');else print(x);putchar('\n');}

#define lf double

const int maxn = 2e5+10;
const int inf = 1e9;
const lf eps = 1e-8;

ll lstans;
struct edge{int to,nxt,w;}e[maxn<<1];

struct data{
ll v;int c;
bool operator < (const data &rhs) const {return c<rhs.c;}
}space[maxn*100];

bool operator < (const int a,data b) {return a<b.c;}
bool operator < (data a,const int b) {return a.c<b;}

data *f[maxn],*s[maxn][3];
int tr[maxn][3];

struct Input_Tree {
ll dist[maxn];
int g[maxn][20],dep[maxn];
void dfs(int x,int Fa) {
g[x][0]=Fa;
for(int i=1;i<=18;i++) g[x][i]=g[g[x][i-1]][i-1];
if(e[i].to!=Fa) dep[e[i].to]=dep[x]+1,dist[e[i].to]=dist[x]+e[i].w,dfs(e[i].to,x);
}
int lca(int x,int y) {
if(dep[x]<dep[y]) swap(x,y);
for(int i=18;~i;i--) if(dep[g[x][i]]>=dep[y]) x=g[x][i];
if(x==y) return x;
for(int i=18;~i;i--) if(g[x][i]!=g[y][i]) x=g[x][i],y=g[y][i];
return g[x][0];
}
ll dis(int x,int y) {return dist[x]+dist[y]-2*dist[lca(x,y)];}
}T;

void get_rt(int x,int Fa) {
sz[x]=1,d[x]=0;
if(e[i].to!=Fa&&!vis[e[i].to]) {
get_rt(e[i].to,x),sz[x]+=sz[e[i].to];
if(d[x]<sz[e[i].to]) d[x]=sz[e[i].to];
}
d[x]=max(d[x],size-sz[x]);
if(d[rt]>d[x]) rt=x;
}

int now;

void dfs(int x,int Fa,ll dis,data *r) {
r[++now]=(data){dis,col[x]};
if(e[i].to!=Fa&&!vis[e[i].to]) dfs(e[i].to,x,dis+e[i].w,r);
}

void solve(int x) {
vis[x]=1,get_rt(x,0),size=sz[x];
f[x]=space+sp;sp+=size+2;now=0;SZ[x]=size;
dfs(x,0,0,f[x]);
sort(f[x]+1,f[x]+size+1);
for(int i=1;i<=size;i++) f[x][i].v+=f[x][i-1].v;
if(!vis[e[i].to]) {
now=0,dfs(e[i].to,x,e[i].w,s[x][cnt]=space+sp),sp+=sz[e[i].to]+3;
if(sz[e[i].to]) {
sort(s[x][cnt]+1,s[x][cnt]+sz[e[i].to]+1);
for(int j=1;j<=sz[e[i].to];j++)
s[x][cnt][j].v+=s[x][cnt][j-1].v;
}
}
if(!vis[e[i].to]) {
get_rt(e[i].to,x);size=sz[e[i].to],rt=0;get_rt(e[i].to,x);
fa[rt]=x;tr[x][cnt]=rt,solve(rt);
}
}

ll calc(data *r,int Sz,int l,int R,int op,ll D) {
int i=upper_bound(r+1,r+Sz+1,R)-r;i--;
if(!i) return 0;
int j=lower_bound(r+1,r+Sz+1,l)-r;j--;
return r[i].v-r[j].v+op*D*(i-j);
}

ll query(int u,int l,int r) {
rt=u;ll res=calc(f[u],SZ[u],l,r,0,0);int pre=u;u=fa[u];
for(;u;pre=u,u=fa[u]) {
res+=calc(f[u],SZ[u],l,r,1,T.dis(u,rt));
for(int i=0;i<3;i++)
if(tr[u][i]==pre) res-=calc(s[u][i],SZ[pre],l,r,1,T.dis(u,rt));
}return res;
}

int main() {