Computer >> 컴퓨터 >  >> 프로그램 작성 >> C++

C++의 배열 변환


초기 배열 arr이 있다고 가정하고 매일 전날의 배열을 사용하여 새 배열을 생성한다고 생각하십시오. i번째 날에 i-1일의 배열에 대해 다음 작업을 수행하여 i일의 배열을 생성합니다. 조건은 다음과 같습니다 -

  • 요소가 왼쪽 및 오른쪽 인접 값보다 작으면 이 요소가 증가합니다.

  • 요소가 왼쪽 및 오른쪽 인접 값보다 크면 이 요소가 감소합니다.

  • 첫 번째 요소와 마지막 요소는 동일하게 유지됩니다.

며칠이 지나면 배열이 더 이상 변경되지 않습니다. 그 최종 배열을 찾으십시오. 따라서 초기 배열이 [6,2,3,4]이면 출력은 [6,3,3,4]입니다. 첫날에는 배열이 [6,2,3,4]에서 [6,3,3,4]로 변경되고 더 이상 작업이 수행되지 않습니다.

이 문제를 해결하기 위해 다음 단계를 따릅니다. −

  • 배열 크기가 2 이하이면 배열을 반환합니다.
  • 플래그 설정 :=true
  • 플래그가 참인 동안 -
    • 플래그 설정 :=false
    • temp라는 빈 배열 하나를 만들고 arr[0]을 temp에 삽입
    • 범위 1에서 배열 크기까지 i의 경우 – 1
      • arr[i]
      • 그렇지 않으면 arr[i]> arr[i - 1] 및 arr[i]> arr[i + 1]일 때 arr[i] – 1을 temp에 삽입하고 플래그를 설정합니다.>
      • 그렇지 않으면 temp에 arr[i] 삽입
    • arr의 마지막 요소를 temp에 삽입
    • arr :=임시
  • 반환

예(C++)

더 나은 이해를 위해 다음 구현을 살펴보겠습니다. −

#include <bits/stdc++.h>
using namespace std;
#define push push_back
void print_vector(vector<auto> v){
   cout << "[";
   for(int i = 0; i<v.size(); i++){
      cout << v[i] << ", ";
   }
   cout << "]"<<endl;
}
class Solution {
public:
   vector<int> transformArray(vector<int>& arr) {
      if(arr.size()<=2)return arr;
      bool flag = true;
      while(flag){
         flag = false;
         vector <int> temp;
         temp.push_back(arr[0]);
         for(int i = 1; i < arr.size()-1; i++){
            if(arr[i]< arr[i-1] && arr[i]<arr[i+1]){
               temp.push(arr[i]+1);
               flag = true;
            }
            else if(arr[i]> arr[i-1] && arr[i]>arr[i+1]){
               flag = true;
               temp.push(arr[i]-1);
            }
            else temp.push(arr[i]);
         }
         temp.push_back(arr[arr.size()-1]);
         arr = temp;
      }
      return arr;
   }
};
main(){
   Solution ob;
   vector<int> v = {1,6,3,4,3,5};
   print_vector(ob.transformArray(v));
}

입력

[1,6,3,4,3,5]

출력

[1,4,4,4,4,5]