n개의 피크가 있는 순열을 찾는 C++ 프로그램

두 개의 숫자 n과 k가 있다고 가정합니다. 정확히 k개의 피크를 갖는 1에서 n까지의 숫자를 사용하여 순열 A를 구성해야 합니다. 인덱스 i는 A[i]> A[i-1]이고 A[i]> A[i+1]인 경우 배열 A의 피크라고 합니다. 이것이 불가능하면 -1을 리턴하십시오.

따라서 입력이 n =5와 같으면; k =2이면 출력은 [2, 4, 1, 5, 3]이 되며 다른 답변도 가능합니다.

단계

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

if k > (n - 1) / 2, then:
   return -1
Define an array a of size: 101.
for initialize i := 1, when i <= n, update (increase i by 1), do:
   a[i] := i
for initialize i := 2, when i <= 2 * k, update i := i + 2, do:
   swap a[i] and a[i + 1]
for initialize i := 1, when i <= n, update (increase i by 1), do:
   print a[i]

예시

이해를 돕기 위해 다음 구현을 살펴보겠습니다. −

#include <bits/stdc++.h>
using namespace std;

void solve(int n, int k) {
   if (k > (n - 1) / 2) {
      cout << "-1";
      return;
   }
   int a[101];
   for (int i = 1; i <= n; i++)
      a[i] = i;
   for (int i = 2; i <= 2 * k; i += 2) {
      swap(a[i], a[i + 1]);
   }
   for (int i = 1; i <= n; i++)
      cout << a[i] << ", ";
}
int main() {
   int n = 5;
   int k = 2;
   solve(n, k);
}

입력

5, 2

출력

1, 3, 2, 5, 4,