【算法每日一练]-结构优化(保姆级教程 篇5 树状数组)POJ3067日本 #POJ3321苹果树 #POJ2352星星 #快排变形

2023-12-13 04:45:25

目录

今天知识点

求交点转化求逆序对,每次操作都维护一个y点的前缀和

树的变动转化成一维数组的变动,利用时间戳将节点转化成区间

离散化数组来求逆序对数

先将y排序,然后每加入一个就点更新求一次前缀和

POJ3067:日本

????????思路:

POJ3321苹果树:

????????思路:

快排变形:

????????思路:

POJ2352:星星

????????思路:


?????????

????????

POJ3067:日本

东海岸有n个城市,西海岸有m个城市,每个海岸的城市从北到南编号为1,2……,每条高速公路都是直线,连接东西海岸的城市。求公路的交叉点数

输入:
1
3 4 4
1 4
2 3
3 2
3 1

????????
思路:

根据样例画出草图:按照1 4,2 3,3 1,3 2的顺序去画,很容易发现只要出现逆序对就会产生交点。

定义逆序对:(x1,y1)和(x2,y2)为逆序对,则等价于x1<x2且y1<y2。

所以在画2 3时候产生了一个逆序对,画3 1时候产生了2个逆序对,画 3 2时候也产生了两个逆序对。故最终5个交点。

所以这道题就是求逆序对数。

因为比如两边都同时大于或小于。我们先对x排序(若x相等,则y升序),然后按x的顺序检查每条边,统计y的前缀和,因为当前已经连了i条边,那么y的前缀和数就一定是非逆序对数。所以i减去y的前缀和就是逆序对数。

这道题就变成了每次增加一个元素就前一次对应的前缀和问题。因此我们只需要对每个点y维护一个关于y的前缀。每次操作后都要给对应点y加个1
????????

#include <bits/stdc++.h>
using namespace std;
#define maxn 1010
#define maxk 1000010
#define lowbit(x) (x)&(-x)
typedef long long ll;
int c[maxn],kas,n,m,k;
struct edge{int x,y;}e[maxk];

bool cmp(edge a,edge b){
	return a.x<b.x||(a.x==b.x&&a.y<b.y);
}

void add(int i){//加1操作,参数省略
	while(i<=m){
		++c[i];
		i+=lowbit(i);
	}
}

int sum(int i){
	int s=0;
	while(i>0){
		s+=c[i];
		i-=lowbit(i);
	}
	return s;
}

int main(){
	int t;cin>>t;
	while(t--){
		memset(c,0,sizeof(c));//每个样例都要清空一次树状数组。
		scanf("%d%d%d",&n,&m,&k);
		for(int i=0;i<k;i++)
			scanf("%d%d",&e[i].x,&e[i].y);
			sort(e,e+k,cmp);//默认升序
			ll ans=0;
			for(int i=0;i<k;i++){
				ans+=i-sum(e[i].y);//累加逆序对
				add(e[i].y);//加入进去
			}
			printf("Test case %d: %lld\n",++kas,ans);
	}
}

????????

????????

POJ3321苹果树:

????????
一个苹果树上有n个叉,通过分支连接,我们将叉从1到n进行编号,每个叉上最多只会有一个苹果,且苹果树上一开始长满了苹果。
卡卡可能会从树上摘一个苹果,树上的空叉可能又会长出新的苹果。
输入:
第一行n表示叉数。
以下n-1行是两个整数u和v表示之间有叉相连
以下m行表示m条消息
C x表示叉x上的苹果变化了:有过原来有则现在没有,原来没有则现在有了
Q x表示叉x上方子树中的苹果数量,包括x叉上的苹果(如果存在的话)
3
1 2
1 3
3
Q 1
C 2
Q 1

????????
思路:

????????????????

我们先把树倒过来,既然要统计每个节点的变动,每变动一次就统计一次不现实。

那就把树所有节点按照dfs顺序映射成一维数组a,再利用时间戳就把求节点孩子问题变成了求时间戳的区间和问题

????????????????
既要统计a的区间和又要考虑到节点的变动,那就创建树状数组c来维护a。节点的变动恰好对应了点更新。

红色代表L,蓝色代表R, 可见每个点的时间戳,不难看出每个节点的R-L就是这个节点的孩子数量

#include <bits/stdc++.h>
#define lowbit(x) (x)&(-x)//求区间长度
using namespace std;
const int maxn=1e5+10;
int n,q;
int c[maxn],a[maxn];
int L[maxn],R[maxn];
int head[maxn];
int cnt,dfn;
struct edge{int u,v,next;}e[2*maxn];

void adde(int u,int v){e[++cnt]={u,v,head[u]};head[u]=cnt;}

int sum(int i){//求前缀和,
	int ans=0;
	for(;i>0;i-=lowbit(i)) ans+=c[i];
	return ans;
}

void add(int i,int val){//在第i点上加val,修改找后继
	for(;i<=n;i+=lowbit(i)) c[i]+=val;
}

void init(){
	memset(c,0,sizeof(c));
	memset(L,0,sizeof(L));
	memset(R,0,sizeof(R));
	memset(head,0,sizeof(head));
	cnt=0;dfn=0;//因为深度优先的序列是从1开始的
	for(int i=1;i<=n;i++)a[i]=1,add(i,1);//a[i]是1表示该分支i上有苹果
}

void dfs(int u,int fa){//之所以写fa,是防止走父子边,这样子的话vis就不再需要了
	L[u]=++dfn;//相当于是时间戳,根节点是1
	for(int i=head[u];i;i=e[i].next){
		int v=e[i].v;
		if(v==fa)continue;
		dfs(v,u);
	}
	R[u]=dfn;//记录时间戳
}

int main(){
	cin>>n;
	int u,v;
	init();
	for(int i=1;i<n;i++){
		scanf("%d%d",&u,&v);
		adde(u,v);
	}	
	dfs(1,1);
	cin>>q;
	char op[10];//之所以定义字符串,就是因为字符型于回车不兼容,所以换成字符串输入不怕回车
	for(int i=1;i<=q;i++){
		getchar();
		scanf("%s %d",op,&v);//不用考虑回车问题
		if(op[0]=='C'){
			if(a[L[v]]) add(L[v],-1);
			else add(L[v],1);
			a[L[v]]^=1;//0变1,1变0
		}
		else{
			int s1=sum(R[v]);
			int s2=sum(L[v]-1);
			printf("%d\n",s1-s2);
		}
	}
}

????????

??????

快排变形:

有n个数,每次通过临项交换来数组中的元素变成升序排列,问需要经过多少次交换?
输入5 4 1 2 999999999 ? 输出5

????????
思路:

就是求逆序对数。输入4时有一个逆序对,再输入1有两个,再输入2有两个,再输入99999999有零个。共五个。
因为我们要在输入数时求小于此数前缀和,可以将数值当成下标存入树状数组来求前缀和,但是数值过大就必须离散化处理,把数值变成排名。
比如:5 4 1 2 999999999 变成4 3 1 2 5即可。

方法:先创建一个排序后的暂存数组,然后按每个数的名次进行赋值对原数组修改即可。
????????

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int maxn=10000;
ll ans,c[maxn];//c[]为树状数组
int n,a[maxn],b[maxn];
int lowbit(int i){	return (-i)&i;}

void add(int i,int z){	for(;i<=n;i+=lowbit(i)) c[i]+=z;}

ll sum(int i){
	ll s=0;
	for(;i>0;i-=lowbit(i)) s+=c[i];
	return s;
}
int main(){
	cin>>n;
	//进行离散化
	for(int i=1;i<=n;i++)cin>>b[i],a[i]=b[i];
	sort(b+1,b+1+n);
	for(int i=1;i<=n;i++)
		a[i]=lower_bound(b+1,b+1+n,a[i])-b;
	//开始求前缀和
	for(int i=1;i<=n;i++){
		ans+=i-sum(a[i])-1;
		add(a[i],1);
	}
	cout<<ans;	
	return 0;
}
/*//离散化实例
#include <bits/stdc++.h>
using namespace std;
int a[100],b[100];

int main(){
	int n;cin>>n;
	for(int i=1;i<=n;i++)cin>>b[i],a[i]=b[i];//实现离散化,把数值变成相对排名,然后数值当成下标存储个数统计前缀和
	sort(b+1,b+1+n);
	for(int i=1;i<=n;i++){
		cout<<a[i]<<' ';
		a[i]=lower_bound(b+1,b+1+n,a[i])-b;
		cout<<a[i]<<' '<<'\n';	
	}
}

5
5 4 1 2 999*/

?????????

?????????

POJ2352:星星

在平面上有n个星星,每颗星星都有自己的坐标。规定星星的等级数为纵横坐标均不超过自己的星星数量(不包括自己),请输出每个级别的星星数量
输入保证y是递增的,且如果y相等,那么x是递增的。
5
1 1
5 1
7 1
3 3
5 5

????????
思路:

看似是二维前缀和,实际上y是排好顺序的,那也就是说只需要按y的顺序计算每个x的前缀和即可。相当于加入一个x就统计一下x的前缀和。
????????

#include<bits/stdc++.h>
using namespace std;
#define maxn 32005
#define lowbit(x) (x)&(-x)
int ans[maxn],c[maxn];
int n;

void add(int i,int val){
	while(i<=maxn){
	c[i]+=val;
	i+=lowbit(i);
	}	
}

int sum(int i){//统计前缀和
	int s=0;
	while(i>0){
		s+=c[i];
		i-=lowbit(i);
	}
	return s;
}
int main(){
	cin>>n;
	int x,y;
	for(int i=0;i<n;i++){
		scanf("%d%d",&x,&y);
		x++;//注意给的坐标x是从0开始的,树状数组的下标必须从0开始,所以都加1
		ans[sum(x)]++;
		add(x,1);//x的数量加1
	}
	for(int i=0;i<n;i++){//一共最多n-1个等级
		printf("%d\n",ans[i]);
	}
}

文章来源:https://blog.csdn.net/m0_69478376/article/details/134863057
本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。