set中的查找操作
P05523. ycz的set
Description
pps就给你出了一道set入门题,他觉得你做出来了就代表你的set真正入门了。
由于pps太神了,所以你根本不敢反驳,只能老老实实地做出这题。而且pps表示,如果你不能在1s之内给出答案,pps将不会保你AK IOI
Format
Input
第一行为n,代表操作的个数
之后的n行,每行两个数opt,x
如果opt=1,那么请你在set中插入数x,如果数列中已经有x了,那么输出"orz pps,the number has appeared"
如果opt=2,那么请你在set中删除数x,如果数列中没有x,那么输出"orz pps,we don't have this number"
如果opt=3,那么请你输出在set中小于x的数的最大值,如果没有小于x的数,那么输出"orz pps,pps AK NOI"
如果opt=4,那么请你输出在set中大于x的数的最小值,如果没有大于x的数,那么输出"orz pps,pps AK IOI"
(输出orz时不需要输出引号).
n<=1e5,x<=1e9
Output
对于每个询问操作,输出对应的答案
Samples
输入数据 1
10
1 5
1 2
2 3
1 8
3 8
4 2
1 9
1 5
2 9
4 9
输出数据 1
orz pps,we don't have this number
5
5
orz pps,the number has appeared
orz pps,pps AK IOI
Sol:此题考察学生对lower_bound(),upper_bound()相关操作的理解与变通
对于此类题,通常可在set中事先加入两个数字代表极小与极大值。
#include<bits/stdc++.h> using namespace std; const int inf=2147483647; set<int>s; int main(){ int n; scanf("%d",&n); s.insert(inf); s.insert(-inf); while(n--) { int opt,x; scanf("%d %d",&opt,&x); if(opt==1) { if(s.count(x)) printf("orz pps,the number has appeared\n"); else s.insert(x); } if(opt==2) { if(!s.count(x)) printf("orz pps,we don't have this number\n"); else s.erase(s.lower_bound(x)); } if(opt==3) //在set中小于x的数的最大值 { auto it=--s.lower_bound(x); if(*it==-inf) //找到的结果为-inf,说明是不存在的 printf("orz pps,pps AK NOI\n"); else printf("%d\n",*it); } if(opt==4) //输出在set中大于x的数的最小值 { auto it=s.upper_bound(x); if(*it==inf) //找到的结果为inf,说明是不存在的 printf("orz pps,pps AK IOI\n"); else printf("%d\n",*it); } } return 0; }
P01465. 邻值查找2
Description
输入一个数列。 就是每插入一个,找到前面插入过的与之差最小的值,将他们的差值加入答案
Format
Input
第一行为正整数N ,接下来的n行每行有一个正整数
N≤32767,每个数字≤1,000,000。
Output
如题
Samples
输入数据 1
6
5
1
2
5
4
6
输出数据 1
12
HINT 结果说明:5+|1-5|+|2-1|+|5-5|+|4-5|+|6-5|=5+4+1+0+1+1=12
Sol:此题与上一题类似,只是注意会涉及一些加减法操作,所以数据类型不要过界了
#include<bits/stdc++.h> using namespace std; const int N=1e6+7; int a[N]; set<int>s; main() { int n,w; long long sum=0; cin>>n; cin>>w; s.insert(1<<30); s.insert(-1<<30);
//这里用正负1<<30代表极大极小值 s.insert(w); sum+=w; for(int i=2;i<=n;i++) { int p; cin>>p; int x=*s.lower_bound(p); int y=*(--s.lower_bound(p)); // cout<<"find it "<<x<<" "<<y<<endl; // cout<<x-p<<" "<<p-y<<endl; s.insert(p); sum+=min((x-p),(p-y)); } cout<<sum<<endl; return 0; }
P05520. 木材仓库
Description
博艾市有一个木材仓库,里面可以存储各种长度的木材,但是保证没有两个木材的长度是相同的。作为仓库负责人,你有时候会进货,有时候会出货,因此需要维护这个库存。有不超过 100000 条的操作:
进货,格式1 Length:在仓库中放入一根长度为 Length(不超过10^9) 的木材。如果已经有相同长度的木材那么输出Already Exist。
出货,格式2 Length:从仓库中取出长度为 Length 的木材。如果没有刚好长度的木材,取出仓库中存在的和要求长度最接近的木材。如果有多根木材符合要求,取出比较短的一根。输出取出的木材长度。如果仓库是空的,输出Empty。
Format
Input
Output
Samples
输入数据 1
7
1 1
1 5
1 3
2 3
2 3
2 3
2 3
输出数据 1
3
1
5
Empty
#include<bits/stdc++.h> using namespace std; int inf=2147483647; set<int> s; int main() { int n; cin>>n; s.insert(inf); s.insert(-inf); for(int i=1;i<=n;i++) { int op,x; cin>>op>>x; if(op==1) { if(s.count(x)) { //cout<<"Empty"<<endl; printf("Already Exist\n"); } else { s.insert(x); } } if(op==2) { if(s.size()==2) { cout<<"Empty"<<endl; continue; } set<int>::iterator head=--s.lower_bound(x); set<int>::iterator tail=s.lower_bound(x); //形成一个head..........x.........tail这样的形态 if(x-(*head)<=(*tail)-x&&(*head)!=-inf) //此处如果找到tail指向的是inf那个值,也是可以的 //因为inf-x的值是大于x减去某个小于x的数字的 //但是head所指向的不能是-inf,否则数据会溢出的 { cout<<*head<<endl; s.erase(head); } else { cout<<*tail<<endl; s.erase(tail); } } } }