LeetCode 5096. 数组变换

Posted 2020-12-01 itdef

地址 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