文章目錄

• • 1. 題目
  • 2. 解題

1. 題目

給你一個已經 排好序 的整數數組 nums 和整數 a、b、c。對于數組中的每一個數 x，計算函數值 f(x) = ax^2 + 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]

來源：力扣（LeetCode）
鏈接：https://leetcode-cn.com/problems/sort-transformed-array
著作權歸領扣網絡所有。商業轉載請聯系官方授權，非商業轉載請注明出處。

2. 解題

• a = 0，函數單調，直接求值，檢查首尾是否升序，降序則進行反轉
• a != 0，找到離拋物線對稱軸最近的點，依距離近的優先，向兩側擴展，最后檢查是否需要反轉

class Solution { public:vector<int> sortTransformedArray(vector<int>& nums, int a, int b, int c) {int n = nums.size();vector<int> ans(n);if(a == 0){for(int i = 0; i < n; ++i)ans[i] = f(nums[i],a,b,c);if(ans[0] > ans[n-1])reverse(ans.begin(),ans.end());return ans;}double axis = -b/(2.0*a);int l = 0, r = 1;if(axis <= nums[0]) l = -1, r = 0;else if(axis >= nums[n-1]) l = n-1, r = n;else{while(nums[l]<axis)l++;l--, r = l+1;}double disl, disr;int i = 0;while(l>=0 || r<n){disl = (l>=0)? fabs(nums[l]-axis) : LONG_MAX;disr = (r<n)? fabs(nums[r]-axis) : LONG_MAX;if(disl < disr){ans[i++] = f(nums[l],a,b,c);l--;} else{ans[i++] = f(nums[r],a,b,c);r++;}}if(ans[0] > ans[n-1])reverse(ans.begin(),ans.end());return ans;}int f(int x, int a, int b, int c){return a*x*x+b*x+c;} };

8 ms 9.3 MB

---

我的CSDN博客地址 https://michael.blog.csdn.net/

長按或掃碼關注我的公眾號（Michael阿明），一起加油、一起學習進步！

總結

以上是生活随笔為你收集整理的LeetCode 360. 有序转化数组（抛物线对称轴）的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。