最近刷到一题数组题,突然发现解法可以参考之前一道数组例题的解法,使得时间复杂度从O(n^2)降到O(n),我还是有些小得意,就分享给大家吧,大家刷题遇到也可以将代码优化。我们先来看灵感来源的那道数组题。
例题一
给你一个整数数组 nums,返回 数组 answer ,其中 answer[i] 等于 nums 中除 nums[i] 之外其余各元素的乘积 。题目数据 保证 数组 nums之中任意元素的全部前缀元素和后缀的乘积都在 32 位 整数范围内。请不要使用除法,且在 O(n) 时间复杂度内完成此题。来源:力扣(LeetCode)(238).除自身外数组的乘积。
暴力方法:
遍历整个数组,求哪个位置的乘积就跳过该位置,每个位置的乘积都要遍历一遍数组,当数组元素为n个时,我们要遍历n^2次,这是非常恐怖的次数,等会对比说明。
优化方法,
我们第一遍先算数组每个元素左边的数的乘积,第二遍遍历算每个元素右边的数的乘积。
int* productExceptSelf(int* nums, int numsSize, int* returnSize)
{
(*returnSize)=numsSize;
int*ret=(int*)calloc(numsSize,sizeof(int));//返回的数组
int i=0;
int sum=1;
for(i=0;i<numsSize;i++)
{
ret[i]=sum;
sum*=nums[i];
sum的值随着循环为1,然后是nums[0],然后是nums[0]*nums[1],
他们分别存在ret[0],ret[1],ret[2],这些值恰好是数组中每个数左边的乘积,
那我们再借鉴这个方法算出每个数右端的值的乘积,合起来就得到每个数除自己外的乘积和。
}
sum=1;
for(i=numsSize-1;i>=0;i--)
{
ret[i]*=sum;
sum*=nums[i];
这次遍历从数组右边开始,数组最右边的数其右边是无值的,我们给它乘以一个1。
sum的值先是1,然后是nums[n-1],nums[n-1]*nums[n-2],
他们分别与ret[n-1],ret[n-2],ret[n-3]相乘,这恰好是每个数的右端数的乘积,
结合上面求得的值,我们就能得出每个数除自身外的数的乘积
}
return ret;
}
例题二:
给你一个整数数组 nums ,请计算数组的 中心下标 。数组 中心下标 是数组的一个下标,其左侧所有元素相加的和等于右侧所有元素相加的和。如果中心下标位于数组最左端,那么左侧数之和视为 0 ,因为在下标的左侧不存在元素。这一点对于中心下标位于数组最右端同样适用。如果数组有多个中心下标,应该返回 最靠近左边 的那一个。如果数组不存在中心下标,返回 -1
来源:力扣(LeetCode)
刚开始练题的我一开始只能想到暴力遍历,我们再回想上题,上题分别求得左边乘积和右边乘积,那我们换成加号应该就能求出左边数之和和右边数之和,而且时间复杂度仅仅为0(N),n^2和n的区别:我举例描述:当n等于1000,后者只需要遍历2000次,而前者是需要百万次的遍历,有时不得感叹,这些优化代码是一件迷人的事情。
int pivotIndex(int* nums, int numsSize)
{
int*p=(int*)malloc(sizeof(int)*numsSize);
int i=0;
int num=0;
for(i=numsSize-1;i>=0;i--)
{
p[i]=num;//p依次存放0,nums[n-1],nums[n-1]+nums[n-2]...这恰好是一个数字的右值
num+=nums[i];
}
num=0;
for(i=0;i<numsSize;i++)
{
if(p[i]==num)//num依次为0,nums[0],nums[0]+nums[1]...这恰好是一个数字的左值
分别和先前存入的左值比较,相等则返回下标,中心下标找到了
return i;
num+=nums[i];
}
free(p);
p=NULL;
return -1;
}
如果同学们有任何问题可私信讨论,谢谢大家支持!