bzoj3757【树上莫队】

好久之前的代码，现在题已经交不了辣

#include<iostream>
#include<algorithm>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<cmath>
using namespace std;
typedef long long LL;
inline int read()
{
    int f=0,x=0;char c=getchar();
    for (;c<'0'||c>'9';c=getchar()) f=c=='-'?1:0;
    for (;c>='0'&&c<='9';c=getchar()) x=x*10+c-'0';
    return f?-x:x;
}
const int N=50010,M=100010,P=20;
int n,m,ll=1,to[M],next[M],head[N],col[N],clk,ans[N];
int B,nb,bnum[N],dfn[N],fa[N][P],dis[N],cnt[N],rec=0;
bool vis[N];
struct na
{
    int id,u,v,a,b;
}query[M];
 
bool cmp (const na &a,const na &b)
{
    if (bnum[a.u]==bnum[b.u]) return dfn[a.v]<dfn[b.v];
    return bnum[a.u]<bnum[b.u];
}
 
void inline adde(int u,int v)
{
    to[++ll]=v;next[ll]=head[u];head[u]=ll;
    to[++ll]=u;next[ll]=head[v];head[v]=ll;
}
 
int homeless[N],nhl;
int dfs(int u)
{
    dfn[u]=++clk;
    int sum=0;
    for (int i=head[u];i;i=next[i]) if (to[i]!=fa[u][0])
    {
        dis[to[i]]=dis[u]+1;
        fa[to[i]][0]=u;
        sum+=dfs(to[i]);
        if (sum>=B) 
        {
            nb++;
            while (sum--) 
                bnum[homeless[nhl--]]=nb;
        }
    }
    homeless[++nhl]=u;
    return ++sum;
}
 
void split()
{
    B=(int)sqrt(n);
    dfs(1);
    while (nhl)
        bnum[homeless[nhl--]]=nb;
    for (int i=1;i<P;i++)
        for (int j=1;j<=n;j++) fa[j][i]=fa[fa[j][i-1]][i-1];
}
 
int lca(int u,int v)
{
    if (dis[u]<dis[v]) swap(u,v);
    if (dis[u]>dis[v])
    {
        int del=dis[u]-dis[v];
        for (int i=P-1;~i;i--) if (del&(1<<i)) u=fa[u][i];
    }
    if (u!=v)
    {
        for (int i=P-1;~i;i--) 
            if (fa[u][i]!=fa[v][i]) u=fa[u][i],v=fa[v][i];
        if (u!=v) u=fa[u][0],v=fa[v][0];
    }
    return u;
}
 
void verxor(int u)
{
    if (vis[u]) 
    {
        cnt[col[u]]--;
        if (!cnt[col[u]]) rec--;
    }
    else
    {
        if (!cnt[col[u]]) rec++;
        cnt[col[u]]++;
    }
    vis[u]^=1;
}
 
void pathxor(int u,int v)
{
    int r=lca(u,v);
    while (u!=r) verxor(u),u=fa[u][0];
    while (v!=r) verxor(v),v=fa[v][0];
}
 
void solve()
{
    sort(query+1,query+m+1,cmp);
    int u=1,v=1;
    for (int i=1,r;i<=m;i++)
    {
        pathxor(u,query[i].u);
        pathxor(v,query[i].v);
        verxor(r=lca(query[i].u,query[i].v));
        int &dg=ans[query[i].id];
        dg=rec;
        if (r==1) dg--;
        if (query[i].a!=query[i].b&&cnt[query[i].a]&&cnt[query[i].b]) dg--;
        verxor(r);
        u=query[i].u;v=query[i].v;
    }
}
 
int main()
{
    n=read()+1,m=read();
    for (int i=2;i<=n;i++) col[i]=read();
    for (int i=1,u,v;i<n;i++) u=read()+1,v=read()+1,adde(u,v);
    split();
    for (int i=1,u,v,a,b;i<=m;i++)
    {
        u=read()+1,v=read()+1,a=read(),b=read();
        if (bnum[u]>bnum[v]) swap(u,v);
        query[i]=(na){i,u,v,a,b};
    }
    solve();
    for (int i=1;i<=m;i++) printf("%d\n",ans[i]);
    return 0;
}