截取网站流量,曲阳网站制作公司,赛博网站建设四川,自己有服务器怎么搭建网站传送门
题意#xff1a; 思路#xff1a; 很明显子树问题会想到树启#xff0c;让后如何updateupdateupdate呢#xff1f;一个显然的思路就是维护一个树状数组#xff0c;查询次数kjk_jkj的个数。但是这样复杂度是O(nlog2n)O(nlog^2n)O(nlog2n)的#xf…传送门
题意 思路 很明显子树问题会想到树启让后如何updateupdateupdate呢一个显然的思路就是维护一个树状数组查询次数kjk_jkj的个数。但是这样复杂度是O(nlog2n)O(nlog^2n)O(nlog2n)的有没有更优的方式呢注意到我们可以维护一个出现次数的前缀和记数组cnt[i]cnt[i]cnt[i]为出现次数为iii的颜色个数c[i]c[i]c[i]为iii颜色出现的次数当前颜色为colcolcol。每次更新的时候比如说现在tagtagtag为111按正常的思路应该是c[col],cnt[c[col]−1]−−,cnt[c[col]]c[col],cnt[c[col]-1]--,cnt[c[col]]c[col],cnt[c[col]−1]−−,cnt[c[col]]。这样写的话就需要用树状数组查询一下kkk的个数了但是我们修改一下cnt[i]cnt[i]cnt[i]的含义改成出现次数iii的颜色个数(我语文不好别打脸)这样修改的时候只需要c[col],cnt[c[col]]c[col],cnt[c[col]]c[col],cnt[c[col]]这样更新的时候答案就是cnt[k]cnt[k]cnt[k]了。 复杂度O(nlogn)O(nlogn)O(nlogn)。 如果改成出现次数kjk_jkj的颜色个数我们只需要再维护一下子树中颜色的个数sumsumsum那么anssum−cnt[kj]anssum-cnt[k_j]anssum−cnt[kj]。
//#pragma GCC optimize(2)
#includecstdio
#includeiostream
#includestring
#includecstring
#includemap
#includecmath
#includecctype
#includevector
#includeset
#includequeue
#includealgorithm
#includesstream
#includectime
#includecstdlib
#define X first
#define Y second
#define L (u1)
#define R (u1|1)
#define pb push_back
#define mk make_pair
#define Mid (tr[u].ltr[u].r1)
#define Len(u) (tr[u].r-tr[u].l1)
#define random(a,b) ((a)rand()%((b)-(a)1))
#define db puts(---)
using namespace std;//void rd_cre() { freopen(d://dp//data.txt,w,stdout); srand(time(NULL)); }
//void rd_ac() { freopen(d://dp//data.txt,r,stdin); freopen(d://dp//AC.txt,w,stdout); }
//void rd_wa() { freopen(d://dp//data.txt,r,stdin); freopen(d://dp//WA.txt,w,stdout); }typedef long long LL;
typedef unsigned long long ULL;
typedef pairint,int PII;const int N1000010,mod1e97,INF0x3f3f3f3f;
const double eps1e-6;int n,m;
int col[N],se[N],son[N];
int cnt[N],c[N],ans[N];
vectorintv[N];
vectorPIIquery[N];void dfs_son(int u,int f)
{se[u]1;for(auto x:v[u]) if(x!f) { dfs_son(x,u); se[u]se[x]; if(se[x]se[son[u]]) son[u]x; }
}void update(int u,int f,int tag,int p)
{if(tag1) c[col[u]],cnt[c[col[u]]];else c[col[u]]--,cnt[c[col[u]]1]--;for(auto x:v[u]) if(x!fx!p) update(x,u,tag,p);
}void dfs(int u,int f,int op)
{for(auto x:v[u]) if(x!fx!son[u]) dfs(x,u,0);if(son[u]) dfs(son[u],u,1);update(u,f,1,son[u]);for(int i0;iquery[u].size();i) ans[query[u][i].X]cnt[query[u][i].Y];if(!op) update(u,f,-1,0);
}int main()
{
// ios::sync_with_stdio(false);
// cin.tie(0);scanf(%d%d,n,m);for(int i1;in;i) scanf(%d,col[i]);for(int i1;in-1;i){int a,b; scanf(%d%d,a,b);v[a].pb(b); v[b].pb(a);}for(int i1;im;i){int v,k; scanf(%d%d,v,k);query[v].pb({i,k});}dfs_son(1,0); dfs(1,0,0);for(int i1;im;i) printf(%d\n,ans[i]);return 0;
}
/**/
