LeetCode 1243 數組變換

地址 https://leetcode-cn.com/contest/biweekly-contest-12/problems/array-transformation/

首先，給你一個初始數組 arr。而後，天天你都要根據前一天的數組生成一個新的數組。

第 i 天所生成的數組，是由你對第 i-1 天的數組進行以下操做所得的：

假如一個元素小於它的左右鄰居，那麼該元素自增 1。
假如一個元素大於它的左右鄰居，那麼該元素自減 1。
首、尾元素 永不 改變。
過些時日，你會發現數組將會再也不發生變化，請返回最終所獲得的數組。

例 1：

輸入：[6,2,3,4]
輸出：[6,3,3,4]
解釋：
第一天，數組從 [6,2,3,4] 變爲 [6,3,3,4]。
沒法再對該數組進行更多操做。
示例 2：

輸入：[1,6,3,4,3,5]
輸出：[1,4,4,4,4,5]
解釋：
第一天，數組從 [1,6,3,4,3,5] 變爲 [1,5,4,3,4,5]。
次日，數組從 [1,5,4,3,4,5] 變爲 [1,4,4,4,4,5]。
沒法再對該數組進行更多操做。

算法1
這道題 只是勉強作出來 優化的位置還有不少
徹底的模擬題意
每次循環開了一個數組 和詢問數組同樣的長的數組 依次檢測每一個元素 若是符合題意就在新開的數組裏記錄+1或者-1
最後把新開數組加進原數組
若是新開數組所有爲零 那麼就是沒有變化 就 能夠跳出循環

 1 class Solution {
 2 public:
 3     vector<int> transformArray(vector<int>& arr) {
 4         if(arr.size() == 1 || arr.size() == 2) return arr;
 5 
 6         while(1){
 7             vector<int> addvec(arr.size(),0);
 8             for(int i = 1;i < arr.size()-1;i++){
 9                 if(arr[i] > arr[i-1] && arr[i]>arr[i+1]) addvec[i] = -1;
10                 else if(arr[i] < arr[i-1] && arr[i]<arr[i+1]) addvec[i] = 1;
11             }
12             int isAllZero = true;
13             for(int i = 0; i < addvec.size();i++){
14                 arr[i] += addvec[i];
15                 if(addvec[i] != 0) isAllZero =false;
16             } 
17             if(isAllZero) break;
18         }
19 
20 
21         return arr;
22     }
23 };

View Code