팀 구성원의 인덱스 시퀀스를 찾는 C++ 프로그램

n개의 요소와 숫자 k가 있는 배열 A가 있다고 가정합니다. 한 반에 n명의 학생이 있습니다. i번째 학생의 등급은 A[i]입니다. 모든 팀 구성원의 평가가 구별되도록 k명의 학생으로 팀을 구성해야 합니다. 불가능하면 "Impossible"을 반환하고, 그렇지 않으면 인덱스 시퀀스를 반환합니다.

따라서 입력이 A =[15, 13, 15, 15, 12]와 같으면; k =3이면 출력은 [1, 2, 5]가 됩니다.

단계

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

Define two large arrays app and ans and fill them with
cnt := 0
n := size of A
   for initialize i := 1, when i <= n, update (increase i by 1), do:
      a := A[i - 1]
      if app[a] is zero, then:
         (increase app[a] by 1)
         increase cnt by 1;
         ans[cnt] := i
if cnt >= k, then:
   for initialize i := 1, when i <= k, update (increase i by 1), do:
      print ans[i]
Otherwise
   return "Impossible"

예시

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

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

void solve(vector<int> A, int k) {
   int app[101] = { 0 }, ans[101] = { 0 }, cnt = 0;
   int n = A.size();
   for (int i = 1; i <= n; i++) {
      int a = A[i - 1];
      if (!app[a]) {
         app[a]++;
         ans[++cnt] = i;
      }
   }
   if (cnt >= k) {
      for (int i = 1; i <= k; i++)
         cout << ans[i] << ", ";
   }
   else
      cout << "Impossible";
}
int main() {
   vector<int> A = { 15, 13, 15, 15, 12 };
   int k = 3;
   solve(A, k);
}

입력

{ 15, 13, 15, 15, 12 }, 3

출력

1, 2, 5,