LeetCode 360. Sort Transformed Array

/*
  Given a sorted array of integers nums and integer values a, b and c.
  Apply a function of the form f(x) = ax^2 + bx + c to each element x in
  the array. The return array must be in sorted order.
  Expected time complexity: O(n)
  example:
  nums = [-4, -2, 2, 4], a = 1, b = 3, c = 5
  result = [3, 9, 15, 33]
*/
#include <vector>
#include <climits>
#include <algorithm>
#include <iostream>
using namespace std;

vector<int> sortTransformedArray(vector<int>& nums, int a, int b, int c) {
  vector<int> res;
  for(int i = 0; i < nums.size(); ++i) {
    int sum = a * nums[i] * nums[i] + b * nums[i] + c;
    res.push_back(sum);
  }
  sort(res.begin(), res.end());
  return res;
}
// a variation of merge sort.
vector<int> sortTransformedArrayII(vector<int>& nums, int a, int b, int c) {
  vector<int> res;
  int minValue = INT_MAX;
  int minIndex = 0;
  for(int i = 0; i < nums.size(); ++i) {
    int tmp = a * nums[i] * nums[i] + b * nums[i] + c;
    if(tmp < minValue) {
      minValue = tmp;
      minIndex = i;
    }
  }
  int front = minIndex;
  int end = minIndex + 1;
  while(front >= 0 && end < nums.size()) {
    int front_value = a * nums[front] * nums[front] + b * nums[front] + c;
    int end_value = a * nums[end] * nums[end] + b * nums[end] + c;
    if(front_value < end_value) {
      res.push_back(front_value);
      front--;
    } else {
      res.push_back(end_value);
      end++;
    }
  }
  while(front >= 0) {
    int front_value = a * nums[front] * nums[front] + b * nums[front] + c;
    res.push_back(front_value);
    front--;
  }
  while(end < nums.size()) {
    int end_value = a * nums[end] * nums[end] + b * nums[end] + c;
    res.push_back(end_value);
    end++;
  }
  return res;
}

int main(void) {
  vector<int> nums {-4, -2, 2, 4};
  vector<int> res = sortTransformedArrayII(nums, -1, 3, 5);
  for(int i = 0; i < res.size(); ++i) {
    cout << res[i] << endl;
  }
}

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值