例题:给一棵n个节点的树,每个节点有一个颜色,q次查询颜色c在子树i上出现次数。
首先我们可以快速想到一种时间复杂度为的暴力做法。为啦和启发式合并相关,我们考虑这样的暴力算法。
void dfs(int x,int fa)
{
for(int i=0;i<g[x].size();i++)
{
int y=g[x][i];
if(y==fa)continue;
dfs(y,x);
}
memset(cnt,0,sizeof(cnt));
for(int i=st[x];i<=ed[x];i++) //st[x] ed[x]存的是dfs序。这里可以O(n)访问到x的所有子孙。
{
int y=be[i];
cnt[c[y]]++;
}
//记录答案
}
对于一个节点,我们定义其最大子树为重儿子,其它皆为轻儿子。然后,在更新时,如果每个节点都继承其重儿子的答案,就可以减少更新,然后就可以将时间复杂度降为为。这个证明,容我想想。
void build(int x)
{
sz[x]=1;
st[x]=++top;
be[top]=x;
for(int i=0;i<g[x].size();i++)
{
int y=g[x][i];
build(y);
sz[x]+=sz[y];
}
ed[x]=top;
}
void dfs(int x,int fa,int op)
{
int bc=0;
for(int i=0;i<g[x].size();i++)
{
int y=g[x][i];
if(y==fa)continue;
if(sz[y]>sz[bc])bc=y;
}
for(int i=0;i<g[x].size();i++)
{
int y=g[x][i];
if(y==fa||y==bc)continue;
dfs(y,x,0);
}
if(!bc)dfs(bc,x,1);
for(int i=0;i<g[x].size();i++)
{
int y=g[x][i];
if(y==fa||y==bc)continue;
for(int j=st[y];j<=ed[y];j++)
{
z=be[j];
cnt[c[z]]++;
}
}
cnt[c[x]]++;
ans[u]=???;
if(!op) //op 为0 将数组清零
for(int j=st[x];j<=ed[x];j++)
cnt[c[i]]=0;
}
题意:给一棵n个节点的树,根节点为1,每个节点都有一个值val[i](val[i]<1e5)。然后求对于x,求这棵树上两点的LCA为x的所有数对中GCD最大为多少。
题解:由于val[i]很小,每个数约数很少,不超过200,然后只要数上启发式合并记录一下这个约数在这个子树上是否出现过,就可以求出答案。访问到一个节点,首先从重儿子那继承啦约数,由于要求LCA为根节点,只要比较重儿子和根,然后合并,然后依次和每一个轻儿子比较,就可以求出答案,维护过程可以用bitset维护。
#include<bits/stdc++.h>
using namespace std;
const int N=1e5+1;
const int T=1e5;
int sz[N],st[N],ed[N],be[N],va[N],top=0,ans[N];
vector<int> g[N],dv[N];
void build(int x)
{
sz[x]=1;
st[x]=++top;
be[top]=x;
for(int i=0;i<g[x].size();i++)
{
int y=g[x][i];
build(y);
sz[x]+=sz[y];
}
ed[x]=top;
}
bitset<N> btem,bt;
void dfs(int x,int op)
{
int bc=0;
for(int i=0;i<g[x].size();i++)
{
int y=g[x][i];
if(sz[y]>sz[bc])bc=y;
}
for(int i=0;i<g[x].size();i++)
{
int y=g[x][i];
if(y==bc)continue;
dfs(y,0);
}
if(bc)dfs(bc,1);
int z=va[x];
btem.reset();
for(int i=0;i<dv[z].size();i++)
{
btem[dv[z][i]]=true;
}
int k=(btem&bt)._Find_first();
bt|=btem;
for(int i=0;i<g[x].size();i++)
{
int y=g[x][i];
if(y==bc)continue;
btem.reset();
for(int j=st[y];j<=ed[y];j++)
{
z=va[be[j]];
for(int p=0;p<dv[z].size();p++)
{
btem[dv[z][p]]=true;
}
}
k=min(k,(int)(btem&bt)._Find_first());
bt|=btem;
}
ans[x]=T-k;
if(!op)
bt.reset();
}
int main()
{
for(int i=1;i<N;i++)
for(int j=i;j<N;j+=i)
dv[j].push_back(T-i);
int n,x;
scanf("%d",&n);
for(int i=2;i<=n;i++)
{
scanf("%d",&x);
g[x].push_back(i);
}
for(int i=1;i<=n;i++)scanf("%d",&va[i]);
build(1);
dfs(1,1);
for(int i=1;i<=n;i++)
{
printf("%d\n",ans[i]);
}
}