原题链接
#题目描述:
给你一个长度固定的整数数组 arr,请你将该数组中出现的每个零都复写一遍,并将其余的元素向右平移。
注意:请不要在超过该数组长度的位置写入元素。
要求:请对输入的数组 就地 进行上述修改,不要从函数返回任何东西。
示例 1:
输入:[1,0,2,3,0,4,5,0]
输出:null
解释:调用函数后,输入的数组将被修改为:[1,0,0,2,3,0,0,4]
示例 2:
输入:[1,2,3]
输出:null
解释:调用函数后,输入的数组将被修改为:[1,2,3]
提示:
1 <= arr.length <= 10000
0 <= arr[i] <= 9
思路一:一看代码,你就懂了
但是性能很垃圾 44ms
class Solution {
public:
void duplicateZeros(vector<int>& arr) {
int len=arr.size();
for(int i=0;i<arr.size();i++){
if(arr[i]==0){
arr.insert(arr.begin()+i,0);
i++;//插入后容器已增加,需要跳过下一个元素,否则将无限插入
}
}
arr.resize(len);//重新分配容器空间,将多余的删去
}
};
思路二:看注释
class Solution {
public:
void duplicateZeros(vector<int>& arr) {
//先将原来的vector拷贝过来
vector<int> arr2(arr);
//记录复写0后个位置的偏移量
//如:1,5,0,6,7,0,8,1
// 1,5,0,0,6,7,0,0,8,1 偏移后
// 0,1,2,3,4,5,6,7,8,9
//5的下标+1,8的下标+2。
//可以发现每出现一个0,就将0后的下标+1,
//记录因为0的出现,导致的每一个元素应该偏移的量dis,记录于num数组。
int dis=0;
vector<int> num;
for(int i=0;i<arr.size();i++){
if(arr[i]==0){
dis++;
}
num.push_back(dis);
}
//将arr全部清0;
arr.assign(arr.size(),0);
//偏移量作用。
for(int i=0;i+num[i]<arr2.size();i++){
arr[i+num[i]]=arr2[i];
}
}
};
作者:waker-3
链接:https://leetcode-cn.com/problems/duplicate-zeros/solution/cbian-jie-zhi-dao-by-waker-3/
来源:力扣(LeetCode)
著作权归作者所有。商业转载请联系作者获得授权,非商业转载请注明出处。