C语言——双指针法求有序数组的平方
代码
#include <stdio.h>
#include <stdlib.h>
int* sortedSquares(int* nums, int numsSize)
{
int left = 0;
int right = numsSize - 1;
int* ans = (int*)malloc(sizeof(int) * numsSize);
int index;
for (index = numsSize - 1; index >= 0; index--)
{
int lsquare = nums[left] * nums[left];
int rsquare = nums[right] * nums[right];
if (lsquare > rsquare)
{
ans[index] = lsquare;
left++;
}
else
{
ans[index] = rsquare;
right--;
}
}
for (int i = 0; i < numsSize; i++)
{
printf("%d ", ans[i]);
}
free(ans);
}
int main()
{
int numsSize;
scanf_s("%d", &numsSize);
int* nums = (int*)malloc(sizeof(int) * numsSize);
if (nums == NULL)
{
return 1;
}
else
{
for (int i = 0; i < numsSize; i++)
{
scanf_s("%d", nums + i);
}
}
sortedSquares(nums, numsSize);
free(nums);
return 0;
}
结果
