題3：

以下程式碼有什麼問題？如何修改？【中國某著名綜合軟體公司2005年面試題】

// P96_example3.cpp : Defines the entry point for the console application.
//

#include "stdafx.h"
#include <iostream>
#include <vector>

void print(std::vector<int>);


int _tmain(int argc, _TCHAR* argv[])
{
	std::vector<int> vec;
	vec.push_back(1);
	vec.push_back(6);
	vec.push_back(6);
	vec.push_back(3);
	//刪除vec陣列中的所有6
	std::vector<int>::iterator itor1;
	std::vector<int>::iterator itor2;
	for(itor1 = vec.begin(); itor1 != vec.end(); itor1++)
	{
		if(6 == *itor1)
		{
			itor2 = itor1;
			vec.erase(itor2);	//刪除指定位置的元素
		}
	}
	print(vec);
	return 0;
}

void print(std::vector<int> v)
{
	std::cout<<"vector size is: "<<v.size()<<std::endl;
	std::vector<int>::iterator p = v.begin();
	while(p != v.end())
	{
		std::cout<<*p<<std::endl;
		p++;
	}
}
 

這是迭代器問題，只能刪除第一個6，以後迭代器就失效了，不能刪除之後的元素。

itor2 = itor1；這句說明兩個迭代器是一樣的。vec.erase(itor2)；等於vec.erase(itor1)；，這時指標已經指向下一個元素了。itor1++；又自增了，指向了下一個元素3，略過了第二個6。

答案：

修改方法1：使用vector模版裡面的remove函式進行修改，程式碼如下：

// P96_example3.cpp : Defines the entry point for the console application.
//

#include "stdafx.h"
#include <iostream>
#include <vector>
#include <algorithm>
 



void print(std::vector<int>);


int _tmain(int argc, _TCHAR* argv[])
{
	std::vector<int> vec;
	vec.push_back(1);
	vec.push_back(6);
	vec.push_back(6);
	vec.push_back(3);
	//刪除vec陣列中的所有6
	std::vector<int>::iterator itor1;
	std::vector<int>::iterator itor2;
	itor1 = vec.begin();
	vec.erase(std::remove(vec.begin(),vec.end(),6),vec.end());
 

	print(vec);
	return 0;
}

void print(std::vector<int> v)
{
	std::cout<<"vector size is: "<<v.size()<<std::endl;
	std::vector<int>::iterator p = v.begin();
	while(p != v.end())
	{
		std::cout<<*p<<std::endl;
		p++;
	}
}

// P96_example3.cpp : Defines the entry point for the console application.
//

#include "stdafx.h"
#include <iostream>
#include <vector>


void print(std::vector<int>);


int _tmain(int argc, _TCHAR* argv[])
{
	std::vector<int> vec;
	vec.push_back(1);
	vec.push_back(6);
	vec.push_back(6);
	vec.push_back(3);
	//刪除vec陣列中的所有6
	std::vector<int>::iterator itor1;
	std::vector<int>::iterator itor2;
	itor1 = vec.begin();
	for(itor1 = vec.begin(); itor1 != vec.end(); itor1++)
	{
		if(6 == *itor1)
		{
			itor2 = itor1;
			vec.erase(itor2);	//刪除指定位置的元素
			itor1--;
		}
	}
	print(vec);
	return 0;
}

void print(std::vector<int> v)
{
	std::cout<<"vector size is: "<<v.size()<<std::endl;
	std::vector<int>::iterator p = v.begin();
	while(p != v.end())
	{
		std::cout<<*p<<std::endl;
		p++;
	}
}