[CSP-S模拟测试]:模板(ac)（线段树启发式合并）

题目描述

辣鸡$ljh\ NOI$之后就退役了，然后就滚去学文化课了。
他每天都被$katarina$大神虐，仗着自己学过一些姿势就给$katarina$大神出了一道题。
有一棵$n$个节点的以$1$号节点为根的树，每个节点上有一个小桶，节点$u$上的小桶可以容纳个小球，$ljh$每次可以给一个节点到根路径上的所有节点的小桶内放一个小球，如果这个节点的小桶满了则不能放进这个节点，在放完所有小球之后就企图去刁难$katarina$大神，让$katarina$大神回答每个节点的小桶内的小球有多少种颜色。
然而$katarina$大神一眼就秒掉了，还说这就是一道傻逼模板题。
现在$katarina$大神想考考即将参加$NOIP2019$的你能不能回答上辣鸡$ljh$的问题。

输入格式

第一行，一个整数$n$，树上节点的数量。
接下来$n-1$行，每行两个整数$u,v$，表示在$u,v$之间有一条边。
接下来一行$n$个整数，$k_1~k_n$表示每个节点上的小桶数量。
下一行是一个整数$m$，表示$ljh$进行的操作数量。
接下来$m$行，每行两个整数$x,c$，分别表示进行操作的节点和小球颜色。
下一行是一个整数$Q$，表示你需要回答的询问数。
接下来$Q$行，每行一个整数$x$，表示一个询问。

输出格式

对于每个询问输出一行表示这个询问的答案。

样例

样例输入1：

5
1 2
2 3
3 4
2 5
2 1 1 1 1
2
2 1
4 2
3
1
3
5

样例输出1：

2
1
0

样例输入2：

10
3 10
2 5
3 2
2 6
1 9
8 7
7 4
3 8
3 1
15 47 23 22 9 16 45 39 21 13
10
10 7
9 3
5 1
5 2
9 4
10 9
2 4
10 1
2 6
7 9
3
1
2
3

样例输出2：

7
4
6

数据范围与提示

对于$5\%$的数据，$n\leqslant 10,m\leqslant 10,k_i\leqslant 10$。

对于$30\%$的数据，$n\leqslant {10}^3,m\leqslant {10}^3,k_i\leqslant {10}^3$。

对于另外$40\%$的数据，$n\leqslant {10}^5,m\leqslant {10}^5,k_i={10}^5$。

对于$100\%$的数据，$n\leqslant {10}^5,m\leqslant {10}^5,k_i\leqslant {10}^5$。

题解

$5\%$算法：

到现在我都没想出来。

$30\%$算法：

暴力修改，$\Theta(1)$查找答案。

时间复杂度：$\Theta(n^2)$。

期望得分：$30$分。

$40\%$算法：

注意$k_i={10}^5$，也就是说，每一个桶肯定不会装超，问题也就简单了许多，转化成了[BZOJ3307]:雨天的尾巴这道题。

区别在于，这道题是维护总和，而那道题是维护最大值，而且这道题还不用LCA。

时间复杂度：$\Theta(n\log n)$。

期望得分：$40$分。

总得分：$70$分。

$100\%$算法：

对于每一个结点都用一棵线段树维护，下标为时间，存储这个时间子树中是否已经加入了小球和贡献（注意如果之前已经加入了这个小球则贡献为0）。

在线段树上二分查找答案就好了。

但是，显然这样还不够，无论是时间还是空间。

所以我们使用线段树启发式合并，利用启发式合并的方式处理出当前子树中的所有操作，同时构建出新的线段树。

我们还可以先预处理出重儿子，这样我们就可以更少的改变当前线段树里的值。

时间复杂度：$\Theta(n\log^2n)$。

期望得分：$100$分。

代码时刻

$30\%$算法：

#include<bits/stdc++.h>
using namespace std;
struct rec
{int nxt;int to;
}e[200001];
int N,M,Q;
int head[100001],cnt;
int k[100001];
int fa[100001];
bool Map[1001][1001];
int ans[100001];
void add(int x,int y)
{e[++cnt].nxt=head[x];e[cnt].to=y;head[x]=cnt;
}
void pre_dfs(int x)//预处理
{for(int i=head[x];i;i=e[i].nxt)if(e[i].to!=fa[x]){fa[e[i].to]=x;pre_dfs(e[i].to);}
}
void ufs(int x,int c)//疯狂往上找，暴力修改
{while(1){if(!x)break;if(!Map[x][c]&&k[x]>0)ans[x]++;Map[x][c]=1;k[x]--;x=fa[x];}
}
int main()
{scanf("%d",&N);for(int i=1;i<N;i++){int u,v;scanf("%d%d",&u,&v);add(u,v);add(v,u);}for(int i=1;i<=N;i++)scanf("%d",&k[i]);scanf("%d",&M);if(N<=1000&&M<=1000){pre_dfs(1);while(M--){int x,c;scanf("%d%d",&x,&c);ufs(x,c);}scanf("%d",&Q);while(Q--){int x;scanf("%d",&x);printf("%d\n",ans[x]);}}return 0;
}

$40\%$算法：

#include<bits/stdc++.h>
using namespace std;
struct rec
{int nxt;int to;
}e[200001];
struct node
{int x;int c;
}q[100001];
int N,M,Q;
int head[100001],cnt;
int root[100001],trsum[10000001],ls[10000001],rs[10000001],tot;
int k[100001];
int dfsv[100001];
bool vis[100001];
int sum;
int ans[100001];
bool cmp(node a,node b){return a.c<b.c;}
void add(int x,int y)
{e[++cnt].nxt=head[x];e[cnt].to=y;head[x]=cnt;
}
void insert(int &x,int k,int l,int r)
{if(!x)x=++tot;if(l==r){trsum[x]=1;return;}int mid=(l+r)>>1;if(k<=mid)insert(ls[x],k,l,mid);else insert(rs[x],k,mid+1,r);trsum[x]=trsum[ls[x]]+trsum[rs[x]];
}
int merge(int x,int fl,int l,int r)//合并
{if(!x||!fl)return x+fl;if(l==r){trsum[x]=trsum[x]||trsum[fl];return x;}int mid=(l+r)>>1;ls[x]=merge(ls[x],ls[fl],l,mid);rs[x]=merge(rs[x],rs[fl],mid+1,r);trsum[x]=trsum[ls[x]]+trsum[rs[x]];return x;
}
void dfs(int x)
{vis[x]=1;for(int i=head[x];i;i=e[i].nxt)if(!vis[e[i].to]){dfs(e[i].to);root[x]=merge(root[x],root[e[i].to],1,M);}ans[x]=trsum[root[x]];
}
int main()
{scanf("%d",&N);for(int i=1;i<N;i++){int u,v;scanf("%d%d",&u,&v);add(u,v);add(v,u);}for(int i=1;i<=N;i++)scanf("%d",&k[i]);scanf("%d",&M);for(int i=1;i<=M;i++)scanf("%d%d",&q[i].x,&q[i].c);sort(q+1,q+M+1,cmp);for(int i=1;i<=M;i++){if(q[i].c!=q[i-1].c)sum++;//算是一个预处理吧，思路有些清奇insert(root[q[i].x],sum,1,M);}dfs(1);scanf("%d",&Q);while(Q--){int x;scanf("%d",&x);printf("%d\n",ans[x]);}return 0;
}

$100\%$算法：

#include<bits/stdc++.h>
#define L(x) x<<1
#define R(x) x<<1|1
using namespace std;
struct rec
{int nxt;int to;
}e[200001];
struct node
{int x;int c;
}q[100001];
int N,M,Q;
int head[100001],cnt;
int t[100001];
map<int,int> col;//会有负的权值，所以用map
int sum;
bool vis[100001];
int son[100001];
int tr[400001],lz[400001],tle[400001],size[400001];
int ans[100001];
vector<pair<int,int> >super[100001];//用vector存储每一个点内球的颜色和时间，二维数组会MLE
void add(int x,int y)
{e[++cnt].nxt=head[x];e[cnt].to=y;head[x]=cnt;
}
void super_memset(int x)//清空
{tr[1]=size[1]=0;lz[1]=1;for(int i=0;i<super[x].size();i++)tle[super[x][i].first]=0;
}
void pushup(int x)
{tr[x]=tr[L(x)]+tr[R(x)];size[x]=size[L(x)]+size[R(x)];
}
void pushdown(int x)
{if(!lz[x])return;size[L(x)]=size[R(x)]=tr[L(x)]=tr[R(x)]=lz[x]=0;lz[L(x)]=lz[R(x)]=1;
}
void change(int x,int l,int r,int k,int v,int w)
{tr[x]+=v;size[x]+=w;if(l==r)return;pushdown(x);int mid=(l+r)>>1;if(k<=mid)change(L(x),l,mid,k,v,w);else change(R(x),mid+1,r,k,v,w);pushup(x);
}
void add(int x)
{for(int i=0;i<super[x].size();i++){pair<int,int> flag=super[x][i];if(!tle[flag.first]){change(1,1,M,flag.second,1,0);tle[flag.first]=flag.second;}else if(tle[flag.first]>flag.second){change(1,1,M,tle[flag.first],-1,0);change(1,1,M,flag.second,1,0);tle[flag.first]=flag.second;}change(1,1,M,flag.second,0,1);}
}
int ask(int x,int l,int r,int k)
{if(k<=0)return 0;if(l==r)return tr[x];pushdown(x);int mid=(l+r)>>1;if(size[L(x)]<=k)return tr[L(x)]+ask(R(x),mid+1,r,k-size[L(x)]);else return ask(L(x),l,mid,k);
}
void connect(int x,int y)
{for(int i=0;i<super[y].size();i++)super[x].push_back(super[y][i]);super[y].clear();
}
void pre_dfs(int x)
{size[x]=super[x].size();vis[x]=1;for(int i=head[x];i;i=e[i].nxt)if(!vis[e[i].to]){pre_dfs(e[i].to);size[x]+=size[e[i].to];if(size[e[i].to]>size[son[x]])son[x]=e[i].to;}
}
void pro_dfs(int x,int fa)
{for(int i=head[x];i;i=e[i].nxt)if(e[i].to!=fa&&e[i].to!=son[x]){pro_dfs(e[i].to,x);super_memset(e[i].to);}if(son[x])pro_dfs(son[x],x);add(x);for(int i=head[x];i;i=e[i].nxt)if(e[i].to!=fa&&e[i].to!=son[x])add(e[i].to);ans[x]=ask(1,1,M,t[x]);if(son[x]){connect(son[x],x);swap(super[x],super[son[x]]);for(int i=head[x];i;i=e[i].nxt)if(e[i].to!=fa)connect(x,e[i].to);}
}
int main()
{scanf("%d",&N);for(int i=1;i<N;i++){int u,v;scanf("%d%d",&u,&v);add(u,v);add(v,u);}for(int i=1;i<=N;i++)scanf("%d",&t[i]);scanf("%d",&M);for(int i=1;i<=M;i++){scanf("%d%d",&q[i].x,&q[i].c);if(!col[q[i].c]){col[q[i].c]=++sum;q[i].c=sum;}else q[i].c=col[q[i].c];super[q[i].x].push_back(make_pair(q[i].c,i));}pre_dfs(1);memset(size,0,sizeof(size));pro_dfs(1,0);scanf("%d",&Q);while(Q--){int x;scanf("%d",&x);printf("%d\n",ans[x]);}return 0;
}

rp++

转载于:https://www.cnblogs.com/wzc521/p/11271797.html