[分块 阈值 离线] BZOJ 3351 [ioi2009]Regions

这个题啊 自己刚了三小时树分块 没什么卵用
我就说 怎么会有题出树分块

那么的话 我们对询问 $(a,b)$ 按照 b 这种颜色在树上的出现次数分类 阈值设为$\sqrt n$

如果b出现次数 < $\sqrt n$ 那么我们暴力对每个为b的点询问一通
在dfs时边记录边询问可以做到 $O(q\sqrt n)$

然后如果 b出现次数 > $\sqrt n$ 那么这种b最多只有 $\sqrt n$ 种 那么我们暴力对每个为a的点询问一通 每个点最多要询问$\sqrt n$个b 在dfs时边记录边询问可以做到 $O(n\sqrt n)$

时间复杂度 $O(n\sqrt n+q \sqrt n)$

#include<cstdio>
#include<cstdlib>
#include<algorithm>
#include<map>
#include<vector>
#include<cmath>
using namespace std;
typedef long long ll;

inline char nc(){
  static char buf[100000],*p1=buf,*p2=buf;
  return p1==p2&&(p2=(p1=buf)+fread(buf,1,100000,stdin),p1==p2)?EOF:*p1++;
}
inline void read(int &x){
  char c=nc(),b=1;
  for (;!(c>='0' && c<='9');c=nc()) if (c=='-') b=-1;
  for (x=0;c>='0' && c<='9';x=x*10+c-'0',c=nc()); x*=b;
}

const int N=200005;
const int Rr=25005;

struct edge{
  int u,v,next;
}G[N]; int head[N],inum;
inline void add(int u,int v,int p){
  G[p].u=u; G[p].v=v; G[p].next=head[u]; head[u]=p;
}

int n,R,val[N];
int cnt[Rr],tmp[Rr];

typedef pair<int,int> abcd;
vector<abcd> f[Rr],s[Rr];
map<abcd,int> cur; int fat[N];
ll Ans[N];
#define V G[p].v
inline void dfs1(int u){
  int t=val[u];
  tmp[t]++;
  for (int i=0;i<(int)s[t].size();i++)
    Ans[s[t][i].first]+=tmp[s[t][i].second];
  for (int p=head[u];p;p=G[p].next)
    dfs1(V);
  tmp[val[u]]--;
}
inline void dfs2(int u){
  int t=val[u];
  for (int i=0;i<(int)f[t].size();i++)
    Ans[f[t][i].first]-=tmp[f[t][i].second];
  tmp[t]++;
  for (int p=head[u];p;p=G[p].next)
    dfs2(V);
  for (int i=0;i<(int)f[t].size();i++)
    Ans[f[t][i].first]+=tmp[f[t][i].second];
}

int main(){
  int fa,Q,a,b; 
  freopen("line.in","r",stdin);
  freopen("line.out","w",stdout);
  read(n); read(R); read(Q);
  read(val[1]);
  for (int i=2;i<=n;i++) read(fa),read(val[i]),add(fa,i,++inum);
  int B=sqrt(n)+1;
  for (int i=1;i<=n;i++) cnt[val[i]]++;
  for (int i=1;i<=Q;i++){
    read(a); read(b); fat[i]=i;
    if (cur.count(abcd(a,b)))
      fat[i]=cur[abcd(a,b)];
    else{
      cur[abcd(a,b)]=i;
      if (cnt[b]>B) f[a].push_back(abcd(i,b));
      else s[b].push_back(abcd(i,a));
    }
  }
  dfs1(1); dfs2(1);
  for (int i=1;i<=Q;i++)
    printf("%lld\n",Ans[fat[i]]);
  return 0;
}