[蓝桥杯]删除数组中的0元素 （set练习）

[蓝桥杯][算法提高VIP]删除数组中的0元素

时间限制: 1Sec 内存限制: 128MB 提交: 433 解决: 274

题目描述
编写函数CompactIntegers，删除数组中所有值为0的元素，其后元素向数组首端移动。注意，CompactIntegers函数需要接收数组及其元素个数作为参数，函数返回值应为删除操作执行后数组的新元素个数。
输入时首先读入数组长度，再依次读入每个元素。
将调用此函数后得到的数组和函数返回值输出。
输入
无
输出
无
样例输入
7
2 0 4 3 0 0 5
样例输出
2 4 3 5
4

解题思路

set基础入门练习

完整代码

#include<iostream>
#include<vector>
#include<set>
using namespace std;

void showvector(vector<int> a)
{
	vector<int>::iterator it;
	int num=0;
	for(it = a.begin();it != a.end();it++)
	{
		if(*it!=0)
		{
			cout<<*it<<" ";
			num++;
		}
	}
	cout<<endl<<num;
}
int main()
{
	int N;
	cin>>N;
	vector<int>aa;
	for(int i=0;i<N;i++)
	{
		int x;
		cin>>x;
		aa.push_back(x);
	}
	showvector(aa);
}