給你一個已經?排好序?的整數數組?nums?和整數?a、b、c。對于數組中的每一個數 x,計算函數值?f(x) = ax2 + bx + c,請將函數值產生的數組返回。
要注意,返回的這個數組必須按照 升序排列,并且我們所期望的解法時間復雜度為?O(n)。
示例 1:
輸入: nums = [-4,-2,2,4], a = 1, b = 3, c = 5
輸出: [3,9,15,33]
示例 2:
輸入: nums = [-4,-2,2,4], a = -1, b = 3, c = 5
輸出: [-23,-5,1,7]
思路:
先判斷是否為二次函數。
如果是,再判斷開口的上下,根據和對稱軸點的距離來判斷函數值的大小,走雙指針的邏輯。
class Solution {public int[] sortTransformedArray(int[] nums, int a, int b, int c) {int len=nums.length;int[] ans=new int[len];if(a==0){if(b>0){for(int i=0;i<len;i++){ans[i]=b*nums[i]+c;}}else{for(int i=0;i<len;i++){ans[i]=b*nums[len-i-1]+c;}}}else if(a>0){double mid = -b * 1.0 / a / 2;int start=0;int end=len-1;int index=len-1;while(start<=end){if(Math.abs(mid-nums[start])>Math.abs(mid-nums[end])){ans[index--]=a*nums[start]*nums[start]+b*nums[start]+c;start++;}else{ans[index--]=a*nums[end]*nums[end]+b*nums[end]+c;end--;}}}else{double mid = -b * 1.0 / a / 2;int start=0;int end=len-1;int index=0;while(start<=end){if(Math.abs(mid-nums[start])>Math.abs(mid-nums[end])){ans[index++]=a*nums[start]*nums[start]+b*nums[start]+c;start++;}else{ans[index++]=a*nums[end]*nums[end]+b*nums[end]+c;end--;}}}return ans;}
}
?