元素是否在vector中,对vector去重,两个vector求交集、并集
C++虽然对vector封装了大量的函数,比如判断元素是否在vector中find、去重的unique,还有在algorithm类中对两个vector求交集、并集的函数,但是,这些函数是不可以直接vector.XX来使用的,如同在《【C++】容器类》中提到过如何删除vector的其中元素一样,需要搭上迭代器iterator对其进行遍历才能完成这个动作。这是不同于java与c#的,习惯就好。具体如下代码所示:
#include "iostream"
#include "vector"
#include "algorithm" //sort函数、交并补函数
#include "iterator" //求交并补使用到的迭代器
using namespace std;
//打印容器vector
void print_vector(vector<int> v){
if(v.size()>0){
cout<<"{";
for(int i=0;i<int(v.size());i++){
cout<<v<<",";
}
cout<<"\b}";
}
else{
cout<<"{}";
}
}
//容器vector中元素的去重
vector<int> unique_element_in_vector(vector<int> v){
vector<int>::iterator vector_iterator;
sort(v.begin(),v.end());
vector_iterator = unique(v.begin(),v.end());
if(vector_iterator != v.end()){
v.erase(vector_iterator,v.end());
}
return v;
}
//两个vector求交集
vector<int> vectors_intersection(vector<int> v1,vector<int> v2){
vector<int> v;
sort(v1.begin(),v1.end());
sort(v2.begin(),v2.end());
set_intersection(v1.begin(),v1.end(),v2.begin(),v2.end(),back_inserter(v));//求交集
return v;
}
//两个vector求并集
vector<int> vectors_set_union(vector<int> v1,vector<int> v2){
vector<int> v;
sort(v1.begin(),v1.end());
sort(v2.begin(),v2.end());
set_union(v1.begin(),v1.end(),v2.begin(),v2.end(),back_inserter(v));//求交集
return v;
}
//判断vector的某一元素是否存在
bool is_element_in_vector(vector<int> v,int element){
vector<int>::iterator it;
it=find(v.begin(),v.end(),element);
if (it!=v.end()){
return true;
}
else{
return false;
}
}
int main(){
vector<int> v1,v2,v;
v1.push_back(22);v1.push_back(22);v1.push_back(23);v2.push_back(23);v2.push_back(24);
cout<<"v1是否存在1这个元素?"<<is_element_in_vector(v1,1)<<endl;
cout<<"对v1去重:";
v1=unique_element_in_vector(v1);
print_vector(v1);
cout<<endl;
cout<<"求v1与v2的交集:";
v=vectors_intersection(v1,v2);
print_vector(v);
cout<<endl;
cout<<"求v1与v2的并集:";
v=vectors_set_union(v1,v2);
print_vector(v);
return 0;
}
页:
[1]