1.并查集
#include<iostream>
using namespace std;
const int lp=1e6+10;
int p[lp];
int find(int x){
if(p[x]!=x) p[x]=find(p[x]);//如果说,根节点不等于当前点的根节点,那么就寻找当前点的根节点,采用dfs和路径压缩
else return p[x];
}
int main(){
int n,m;//n个点 m次查询
cin>>n>>m;
for(int i=1;i<=n;i++) p[i]=i;//必须有,让他们初始时,自己等于自己的的根节点
for(int i=1;i<=m;i++){
int a,b;
cin>>b>>a;//将b合并到a上,和a合并到b上一样
p[find(a)]=find(b);
}
return 0;
}
- 为什么
find()
函数会有if(p[x]!=x)
,刚开始我们已经将所有的点都定为自己的跟节点,那么当合并第一个联通块的时候,只会执行return
,但是,当第二次合并的时候if
语句就起到了作用;
- 例如 ,合并区间[1,2,3] [4,5,6]不妨令第一个数字为根节点,那么当
a=1 b=5
的时候 ,这里只说b
,p[b]=4
那么p[b]!=b
就会进行dfs
找到4 的位置,他的根节点p[4]=4 那么就return 返回4
,之后p[4] p[5]
的根节点都会被更新为 4 (其实在第二个在第二个区间形成的时候5的根节点已经被更新为4),然后由主函数的p[find(a)]=find(b);
实现了区间合并 - 在第一次合并但是未再次遍历的时候,
[4,5,6]
只有 4的跟节点被更新为 1,但是,等再次访问的时候,5 6
也会更新,这并不影响区间的更新问题
对于函数 int find(int x)
也可采用非递归方法
int find(int x){
int r=x;
while(r!=s[r]) r=s[r];//找到根节点
int i=x;//从开头开始
while(i!=r){//一直到根节点
//更新路上的点的根节点
int j=s[i];
s[i]=r;
i=j;
}
}
2.树状数组
将数组变成树状数组,( logn)
void up(int x,int c){//从x开始,路径上的点都加上c 类似于前缀和
for(int i=x;i<=n;i+=lowbit(i)) t[i]+=c;
}
求x前面的所有数字的和( logn)
int sum(int x){
int res=0;
for(int i=x;i;i-=lowbit(i)) res+=t[i];
return res;
}
非负整数 x 在二进制表示下最低位1及其后面的0构成的数值
lowbit ( 12 ) = lowbit( [ 1100 ] 2 \ {[1100]}_2 [1100]2 )= [ 100 ] 2 \ {[100]}_2 [100]2=4
int lowbit(int x){
return x & -x;
}
===================== 例题 =========================
#include<iostream>
#include<cstring>
#include<algorithm>
using namespace std;
const int lp=1e5+10;
int t[lp];
int n;
int a[lp];
int lowbit(int x){
return x & -x;
}
void up(int x,int c){
for(int i=x;i<=n;i+=lowbit(i)) t[i]+=c;
}
long long sum(int x){
long long res=0;
for(int i=x;i;i-=lowbit(i)) res+=t[i];
return res;
}
int main(){
int m;
cin>>n>>m;
for(int i=1;i<=n;i++) cin>>a[i];
for(int i=1;i<=n;i++){
up(i,a[i]-a[i-1]);
}
for(int i=1;i<=m;i++){
char ch;
cin>>ch;
if(ch=='Q'){
int x;
cin>>x;
cout<<sum(x)<<endl;
}
else{
int x,y,z;
cin>>x>>y>>z;
up(x,z);
up(y+1,-z);
}
}
return 0;
}