C++中vector使用push_back时注意vector的初始化方式

1 出现的问题

vector<int> smallestK(vector<int>& arr, int k)
{
	vector<int> res(k, 0);
	sort(arr.begin(), arr.end());
	for (int i = 0; i < k; i++)
		res.push_back(arr[i]);
	return res;
}

其中arr={3,4,5},k=2, 返回的结果是{0, 0, 3, 4}.
即出现了两个多余的0.

2 问题分析及解决办法

这里res初始化为k个0，也就是初始的容量已经为K了。**这时候再使用push_back就变成了在原来的K个0的基础上往后追加了。**正确写法为：

vector<int> smallestK(vector<int>& arr, int k)
{
	vector<int> res(k, 0);
	sort(arr.begin(), arr.end());
	for (int i = 0; i < k; i++)
		res[i] = arr[i];
	return res;
}