数据结构小总结（成都磨子桥技工学校数据结构前12题）

[pixiv] https://www.pixiv.net/member_illust.php?mode=medium&illust_id=34352147
暑假的作业，颓颓的我总算是写完了

线段树

线段树是一个高级玩意，不仅可以求区间和，区间最大等等的简单问题，灵活运用还有好多变种。自从学了主席树，知道了null自环这种东西后，用在线段树上也是得心应手

c3

给一个长为N的数列，有M次操作，每次操作是以下两种之一：
（1）修改数列中的一个数
（2）求数列中某连续一段的和

赤裸裸的线段树

c4

给一个长为N的数列，有M次操作，每次操作时以下三种之一：
（1）修改数列中的一个数
（2）求数列中某连续一段所有数的两两乘积的和 mod 1000000007
（3）求数列中某连续一段所有相邻两数乘积的和 mod 1000000007

数据剧毒无比，有负数，取模就出问题了。对于区间维护答案，主要就是如何合并区间。操作3好合并，只要记录每个区间的头、尾的数，把左右儿子区间的和加起来，再加上中间两个数的乘积。关键是操作2，要是没有见识过这个脑筋急转弯，我可能一辈子都不会：

给出N个数， 每次可以合并两个数， 合并的代价是两个数的乘积， 合并得到的数是两个数的和。
问最后把所有数合并成一个数的最小代价。 求这个最小代价对10^9+7取模的结果。
N <= 5000000。

题解是：

显然无论怎么合并答案都是一样的， 任意两个数的乘积恰好会对答案贡献一次。
直接搞就好了

于是这道题的操作2就迎刃而解了
由于这道题坑很多，我就不放我wa掉的代码了
大神的AC代码

#include<iostream>
#include<cstdio>
#include<algorithm>
using namespace std;
const int maxn=100005;
const int mod=1000000007;
struct Node{int s1,s2,s3;
}node[maxn<<2];
int a[maxn];
int n,m;
int slow_mult(int a,int p){if(a<0)a=(a+(mod<<1))%mod;if(p<0)p =(p+(mod<<1))%mod;if(a<p)swap(a,p);if(p==0)return 0;if(p==1)return a%mod;int tmp=slow_mult(a,p>>1);if(p&1)return ((tmp<<1)%mod+a%mod)%mod;else return (tmp<<1)%mod;
}
inline void update(int root,int l,int r){node[root].s1=(node[root<<1].s1+node[root<<1|1].s1)%mod;node[root].s2=(node[root<<1].s2+node[root<<1|1].s2)%mod;int m=(l+r)>>1;node[root].s3=((node[root<<1].s3+node[root<<1|1].s3)%mod+slow_mult(a[m],a[m+1]))%mod;
}
void build(int root,int l,int r){if(l==r){node[root].s1=a[l];node[root].s2=slow_mult(a[l],a[r]);node[root].s3=0;return ;}int m=(l+r)>>1;build(root<<1,l,m),build(root<<1|1,m+1,r);update(root,l,r);
}
void modify(int root,int l,int r,int x,int val){if(l==r){node[root].s1=val;node[root].s2=slow_mult(val , val);node[root].s3=0;a[x]=val;return;}int m=(l+r)>>1;if(x<=m)modify(root<<1,l,m,x,val);else modify(root<<1|1,m+1,r,x,val);update(root,l,r);
}
int query1(int root,int l,int r,int x,int y){if(x<=l&&r<=y)return node[root].s1;int m=(l+r)>>1,ret=0;if(x<=m&&l<=y)ret+=query1(root<<1,l,m,x,y);if(y>=m+1&&r>=x)ret+=query1(root<<1|1,m+1,r,x,y);return ret % mod;
}
int query2(int root,int l,int r,int x,int y){if(x<=l&&r<=y)return node[root].s2;int m=(l+r)>>1,ret=0;if(x<=m&&l<=y)ret+=query2(root<<1,l,m,x,y);if(y>=m+1&&r>=x)ret+=query2(root<<1|1,m+1,r,x,y);return ret % mod;
}
int query3(int root,int l,int r,int x,int y){if(x<=l&&r<=y)return node[root].s3;int m=(l+r)>>1,ret=0,flag=1;if(x<=m&&l<=y)ret+=query3(root<<1,l,m,x,y),flag*=-1;if(y>=m+1&&r>=x)ret+=query3(root<<1|1,m+1,r,x,y),flag*=-1;if(flag^1)return ret%mod;else return(ret%mod+slow_mult(a[m],a[m+1]))%mod;
}
void read(int &res){int flag=1;static char ch;while((ch=getchar())<'0'||ch>'9')if(ch=='-')flag=-1;res=ch-48;while((ch=getchar())>='0'&&ch<='9')res=res*10+ch-48;res*=flag;
}
void reads(char &res){static char ch;while((ch=getchar())!='Q'&&ch!='M'&&ch!='A');res=ch;
}
int main(){read(n),read(m);for(int i=1;i<=n;i++)read(a[i]),a[i]%=mod;build(1,1,n);for(int i=1;i<=m;i++){char cmd;int x,y;reads(cmd);read(x),read(y);if(cmd=='M')modify(1,1,n,x,y%mod);else if(cmd=='Q'){int t1=query1(1,1,n,x,y);int t2=query2(1,1,n,x,y);t1=slow_mult(t1,t1);int tmp=t1-t2;if(tmp<0||tmp&1)tmp+=mod;if(tmp&1)tmp+=mod;printf("%d\n",(tmp>>1)%mod);}else printf("%d\n",query3(1,1,n,x,y));}return 0;
}

c5

给一个长为N的数列，有M次操作，每次操作是以下两种之一：
（1）将某连续一段同时改成一个数
（2）求数列中某连续一段的和

也是很基本的线段树

c6

给一个长为N的数列，有M次操作，每次操作是以下两种之一：
（1）修改数列中的一个数
（2）求数列中有多少个数比它前面的数都大

其实是裸的楼房重建，线段树代码见这里
然后听说这道题用分块也能过，为了练习练习，这次就用分块了

#include<cstdio>
#include<cstring> 
#include<algorithm>
#include<cmath>
#include<vector> 
using namespace std;const int N=100000+5;
const int B=400;
const int oo=0x7fffffff;int n,m,hh[N];
vector<int> a[B];
int blk,pos[N];void getsort(int x){int maxx=-oo;a[x].clear();for(int i=(x-1)*blk+1;i<=min(n,x*blk);i++){if(hh[i]>maxx){a[x].push_back(i);maxx=hh[i];} }
}
void init(){for(int i=1;i<=pos[n];i++){getsort(i);}
}
int erfen(int maxx,int x){int le=1,ri=a[x].size();while(le<ri){int mid=(le+ri)>>1;if(hh[a[x][mid-1]]<=maxx) le=mid+1;else ri=mid; }if(hh[a[x][le-1]]<=maxx){return 0;} return a[x].size()-le+1;
}
int query(){int ans=0,maxx=-oo;for(int i=1;i<=pos[n];i++){ans+=erfen(maxx,i);maxx=max(maxx,hh[a[i][a[i].size()-1]]);}return ans;
}
int main(){scanf("%d%d",&n,&m);blk=(int)sqrt((double)n);for(int i=1;i<=n;i++){scanf("%d",&hh[i]);pos[i]=(i-1)/blk+1;} init();while(m--){char opt[2];scanf("%s",opt);if(opt[0]=='M'){int x,y;scanf("%d%d",&x,&y);hh[x]=y;getsort(pos[x]);}else{printf("%d\n",query());}}return 0;
}

c7

给一个长为N的数列，有M次操作，每次操作是以下两种之一：
（1）修改数列中的一个数
（2）求数列中某个值出现了多少次

在做c8之前是没有想到用值域线段树的，以为会爆空间（int级别的），就偷了个懒，用map
如何处理值域线段树的空间问题，见c8

#include<cstdio> 
#include<cstring> 
#include<map> 
#include<algorithm> 
using namespace std;const int N=100000+5;map<int,int> mp;
int a[N];int main(){int n,m;scanf("%d%d",&n,&m);for(int i=1;i<=n;i++){scanf("%d",&a[i]);mp[a[i]]++;}while(m--){char opt[2];scanf("%s",opt);if(opt[0]=='M'){int x,y;scanf("%d%d",&x,&y);mp[a[x]]--;mp[y]++;a[x]=y;}else{int x;scanf("%d",&x);printf("%d\n",mp[x]);}}return 0;
}

c8

给一个长为N的数列，有M次操作，每次操作是以下三种之一：
（1）插入一个数，若已存在则忽略
（2）删除一个数，若不存在则忽略
（3）求数列中任意两数之差绝对值的最小值

先不考虑空间问题，我一来就想到值域线段树。计算存在的数与最近的前后两数的差值的最小值。每个区间储存存在的最小数和最大数。合并时，左区间的ans，右区间的ans，左区间最大数和右区间的最小数的差，取min。
那么怎么处理空间大小问题呢？数据是2^31，但N,M的范围是10^5，也就是说最少都有2^31-10^5的数根本和此题无关，是浪费空间。其维护的值都是一样的，那么为什么不指向同一个空间呢？于是就把主席树里的null搬过来，不存在的值就由null代替
然后我猜数据里有负数，结果在划分区间是没注意向上还是向下取整的问题，结果T掉了，全部都加上正无穷就好了。之后又没开longlong，int加爆了。。。orz

#include<cstdio>
#include<cstring>
#include<algorithm>
#define ll long long
using namespace std;const ll oo=2147483646;
const ll N=100000+5;ll n,m;
struct Node{Node *ls,*rs;ll ans,maxn,minn,cnt;
}*root,*null,pool[N*80],*tail=pool;Node *newnode(){Node *rt=++tail;rt->ls=rt->rs=null;rt->ans=oo;rt->maxn=rt->minn=rt->cnt=0;return rt;
}
void update(Node *nd){nd->cnt=nd->ls->cnt+nd->rs->cnt;nd->ans=oo;if(nd->ls->cnt>1) nd->ans=min(nd->ans,nd->ls->ans);if(nd->rs->cnt>1) nd->ans=min(nd->ans,nd->rs->ans);if(nd->ls->cnt&&nd->rs->cnt) nd->ans=min(nd->ans,nd->rs->minn - nd->ls->maxn);if(nd->rs->cnt) nd->maxn=nd->rs->maxn;else nd->maxn=nd->ls->maxn;if(nd->ls->cnt) nd->minn=nd->ls->minn;else nd->minn=nd->rs->minn;
}
void insert(Node *&nd,ll le,ll ri,ll pos){if(nd==null) nd=newnode();if(le==ri){nd->cnt=1;nd->ans=oo;nd->maxn=nd->minn=pos;return;}ll mid=(le+ri)/2;if(pos<=mid) insert(nd->ls,le,mid,pos);else insert(nd->rs,mid+1,ri,pos);update(nd);
}
void del(Node *&nd,ll le,ll ri,ll pos){if(nd==null) return ;if(le==ri){nd->cnt=0;nd->ans=nd->maxn=nd->minn=0;return ;}ll mid=(le+ri)/2;if(pos<=mid) del(nd->ls,le,mid,pos);else del(nd->rs,mid+1,ri,pos);update(nd);
}
int main(){null=++tail;null->ls=null->rs=null;null->ans=null->minn=null->maxn=null->cnt=0;root=null;scanf("%lld%lld",&n,&m);ll a;for(ll i=1;i<=n;i++){scanf("%lld",&a);insert(root,0,oo+oo,a+oo);}while(m--){char opt[2];ll x;scanf("%s",opt);if(opt[0]=='I'){scanf("%lld",&x);insert(root,0,oo+oo,x+oo);}else if(opt[0]=='D'){scanf("%lld",&x);del(root,0,oo+oo,x+oo);}else {if(root->cnt>1) printf("%lld\n",root->ans);else printf("-1\n");}}
}

c10

给一个长为N的数列，有M次操作，每次操作是以下两种之一：
（1）修改数列中的一个数
（2）求数列中第K小的值

第k大问题，想来都是线段树。由于每次要修改数据，离散化不太可能，那么就用上文提到的null来节省空间（哎呀我真是太机智了，自己yy出来的诶，也算是进步吧）

#include<cstdio>
#include<cstring>
#include<algorithm>
#define ll long long 
using namespace std;const ll N=100000+5;
const ll oo=2147483646;struct Node {Node *ls,*rs;ll cnt;
}*root,*null,pool[N*80],*tail=pool;
ll n,m,a[N];Node *newnode(){Node *rt=++tail;rt->ls=rt->rs=null;rt->cnt=0;return rt;
}
void modify(Node *&nd,ll le,ll ri,ll pos,ll val){if(nd==null) nd=newnode();if(le==ri){nd->cnt+=val;return ;}ll mid=(le+ri)>>1;if(pos<=mid) modify(nd->ls,le,mid,pos,val);else modify(nd->rs,mid+1,ri,pos,val);nd->cnt=nd->ls->cnt+nd->rs->cnt;
} 
ll query(Node *nd,ll le,ll ri,ll pos){if(le==ri)return le;ll mid=(le+ri)>>1;if(pos<=nd->ls->cnt) return query(nd->ls,le,mid,pos);else return query(nd->rs,mid+1,ri,pos-nd->ls->cnt);
}
int main(){null=++tail;null->ls=null->rs=null;null->cnt=0;root=null;scanf("%lld%lld",&n,&m);for(ll i=1;i<=n;i++){scanf("%lld",&a[i]);modify(root,0,oo+oo,a[i]+oo,1);}while(m--){char opt[2];scanf("%s",opt);if(opt[0]=='Q'){ll x;scanf("%lld",&x);printf("%lld\n",query(root,0,oo+oo,x)-oo);}else {ll x,y;scanf("%lld%lld",&x,&y);modify(root,0,oo+oo,a[x]+oo,-1);modify(root,0,oo+oo,y+oo,1);a[x]=y;}}return 0;
}

总结：
线段树可腻害可腻害了，其实很多的区间问题都可以解决，唯一的关键点就是如何快速的区间合并。俗话说“以不变应万变”，只要解决区间合并的方法就可以了。

主席树

c2

给一个空数列，有M次操作，每次操作是以下三种之一：
（1）在数列后加一个数
（2）求数列中某位置的值
（3）撤销掉最后进行的若干次操作（1和3）

感觉像是数组，又感觉像是主席树。但是数组又不易用指针写，于是一气之下杀鸡用牛刀，用线段树来解决数列问题。。。1A

#include<cstdio> 
#include<cstring> 
#include<algorithm> 
using namespace std;const int N=100000+5;struct Node{Node *ls,*rs;int val;
}*root[N],*null,pool[N*50],*tail=pool;
int len[N],m;Node *newnode(){Node *rt=++tail;rt->ls=rt->rs=null;rt->val=0;return rt;
}
void insert(Node *np,Node *&nd,int le,int ri,int pos,int val){nd=newnode();nd->ls=np->ls,nd->rs=np->rs;nd->val=np->val;if(le==ri){nd->val=val;return ;}int mid=(le+ri)>>1;if(pos<=mid) insert(np->ls,nd->ls,le,mid,pos,val);else insert(nd->rs,nd->rs,mid+1,ri,pos,val);
}
int query(Node *nd,int le,int ri,int pos){if(le==ri) return nd->val;int mid=(le+ri)>>1;if(pos<=mid) return query(nd->ls,le,mid,pos);else return query(nd->rs,mid+1,ri,pos);
}
int main(){null=++tail;null->ls=null->rs=null;null->val=0;root[0]=null;scanf("%d",&m);int cnt=0;while(m--){char opt[2];int x;scanf("%s%d",opt,&x);if(opt[0]=='A'){cnt++;len[cnt]=len[cnt-1]+1;insert(root[cnt-1],root[cnt],1,N,len[cnt],x);}else if(opt[0]=='Q'){printf("%d\n",query(root[cnt],1,N,x));}else{cnt++;root[cnt]=root[cnt-x-1];len[cnt]=len[cnt-x-1]; }}return 0;
}

c11

给一个长为N的数列，有M次操作，每次操作是以下两种之一：
（1）修改数列中的一个数
（2）求某次操作后连续一段的和

裸主席树，就当练手速吧

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;const int N=100000+5;struct Node{Node *ls,*rs;int sum;
}*root[N],*null,pool[N*80],*tail=pool;
int n,m,a[N];Node *newnode(){Node *rt=++tail;rt->ls=rt->rs=null;rt->sum=0;return rt;
}
void build(Node *&nd,int le,int ri){nd=newnode();if(le==ri){nd->sum=a[le];return ;}int mid=(le+ri)>>1;build(nd->ls,le,mid);build(nd->rs,mid+1,ri);nd->sum=nd->ls->sum+nd->rs->sum;
}
void insert(Node *ne,Node *&nd,int le,int ri,int pos,int val){nd=newnode();nd->ls=ne->ls,nd->rs=ne->rs;if(le==ri){nd->sum=val;return ;}int mid=(le+ri)>>1;if(pos<=mid) insert(ne->ls,nd->ls,le,mid,pos,val);else insert(ne->rs,nd->rs,mid+1,ri,pos,val);nd->sum=nd->ls->sum+nd->rs->sum;
}
int query(Node *nd,int le,int ri,int L,int R){if(L<=le&&ri<=R) return nd->sum;int mid=(le+ri)>>1,rt=0;if(L<=mid) rt+=query(nd->ls,le,mid,L,R);if(mid<R) rt+=query(nd->rs,mid+1,ri,L,R);return rt;
}
int main(){null=++tail;null->ls=null->rs=null;null->sum=0;scanf("%d%d",&n,&m);for(int i=1;i<=n;i++) scanf("%d",&a[i]);build(root[0],1,n);for(int i=1;i<=m;i++){char opt[2];scanf("%s",opt);if(opt[0]=='M'){int x,y;scanf("%d%d",&x,&y);insert(root[i-1],root[i],1,n,x,y);}else{int x,y,z;scanf("%d%d%d",&x,&y,&z);printf("%d\n",query(root[z],1,n,x,y));root[i]=root[i-1];}}return 0;
}

c12

给一个长为N的数列，有M次操作，操作仅有一种：
求数列中某连续一段中第K小的值

还是裸的主席树，没错我是来挂代码的

#include<cstdio>
#include<cstring>
#include<algorithm>
#define ll long long 
using namespace std;const ll N=100000+5;
const ll oo=2147483646;struct Node {Node *ls,*rs;ll cnt;
}*root[N],*null,pool[N*100],*tail=pool;
ll n,m,a[N];Node *newnode(){Node *rt=++tail;rt->ls=rt->rs=null;rt->cnt=0;return rt;
}
void insert(Node *ne,Node *&nd,ll le,ll ri,ll pos){nd=newnode();nd->ls=ne->ls,nd->rs=ne->rs;nd->cnt=ne->cnt+1;if(le==ri) return;ll mid=(le+ri)>>1;if(pos<=mid) insert(ne->ls,nd->ls,le,mid,pos);else insert(ne->rs,nd->rs,mid+1,ri,pos);
}
ll query(Node *ne,Node *nd,ll le,ll ri,ll k){if(le==ri) return le;ll mid=(le+ri)>>1;ll lsc=nd->ls->cnt - ne->ls->cnt;if(k<=lsc) return query(ne->ls,nd->ls,le,mid,k);else return query(ne->rs,nd->rs,mid+1,ri,k-lsc);
}
int main(){null=++tail;null->ls=null->rs=null;null->cnt=0;root[0]=null;scanf("%lld%lld",&n,&m);for(ll i=1;i<=n;i++){scanf("%lld",&a[i]);insert(root[i-1],root[i],0,oo+oo,a[i]+oo);}while(m--){ll x,y,z;scanf("%lld%lld%lld",&x,&y,&z);printf("%lld\n",query(root[x-1],root[y],0,oo+oo,z)-oo);}return 0;
}

数组

c0，c1
太简单啦，寒假做的

平衡树

c9

给一个长为N的数列，有M次操作，每次操作是以下两种之一：
（1）删除某个位置的数
（2）求数列某位置的值

一来就想到treap，但是感觉大材小用了，就偷了个懒

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<vector>
using namespace std;vector<int> vec;
int n,m;int main(){scanf("%d%d",&n,&m);for(int i=1;i<=n;i++){int a;scanf("%d",&a);vec.push_back(a);}while(m--){char opt[2];int x;scanf("%s%d",opt,&x);x--;if(opt[0]=='D')vec.erase(vec.begin()+x);else printf("%d\n",vec[x]);}return 0;
}

好啦，就是这样