977 Squares of a Sorted Array

1、第一种方法就是遍历一遍,将每个元素求平均,然后存入数组,利用C++ 自带的排序函数排序。当然,时间耗时最长140ms。

#include <algorithm>

sort(A.begin(), A.end());

2、遍历一遍,获得平方的数组和最后一个负数的坐标。获得负数的坐标neg后,若有非负数,则非负数的坐标从neg+1开始,正数和负数分开处理。,时间耗时104ms。

  vector<int> sortedSquares(vector<int>& A) {
    int neg = -1; 
    for (int i=0; i < A.size(); ++i) {
      if (A[i] < 0) {
        neg = i;
      }   
      A[i] = A[i] * A[i];
    }   
    int pos = neg + 1;
    vector<int> res;
    while (true) {
      if (pos == A.size()) {
        while (neg >= 0) {
          res.push_back(A[neg]);
          --neg;
        }   
        return res;
      }   
      if (neg < 0) {
        while (pos < A.size()) {
          res.push_back(A[pos]);
          ++pos;
        }   
        return res;
      }   
      while (pos < A.size() && A[pos] < A[neg]) {
        res.push_back(A[pos]);
        ++pos;
      }   
      while (pos < A.size() && neg >= 0 && A[pos] >= A[neg]) {
        res.push_back(A[neg]);
        --neg;
      }   
    }   
    // sort(A.begin(), A.end());
  }

3、第一次遍历只遍历到第一个非负数即可,个人感觉应该是耗时最短的,但是和第二种方法耗时一样,104ms。

  vector<int> sortedSquares1(vector<int>& A) {
    int pos = A.size();
    for (int i=0; i < A.size(); ++i) {
      if (A[i] > 0) {
        pos = i;
        break;
      }
    }
    int neg = pos - 1;
    vector<int> res;
    while (true) {
      if (pos == A.size()) {
        while (neg >= 0) {
          res.push_back(A[neg] * A[neg]);
          --neg;
        }
        return res;
      }
      if (neg == -1) {
        while (pos < A.size()) {
          res.push_back(A[pos] * A[pos]);
          ++pos;
        }
        return res;
      }
      while (pos < A.size() && A[pos] * A[pos] < A[neg] * A[neg]) {
        res.push_back(A[pos] * A[pos]);
        ++pos;
      }
      while (pos < A.size() && neg >= 0 && A[pos] * A[pos] >= A[neg] * A[neg]) {
        res.push_back(A[neg] * A[neg]);
        --neg;
      }
    }
  }

 错误记录:error: expected primary-expression before ‘=’ token

将if (pos == A.size()) 错写成if (pos = = A.size())。一般这个错误都是小的语法问题。

猜你喜欢

转载自blog.csdn.net/huanucas/article/details/89003243