สมมติว่าเรามีอาร์เรย์ A ที่มีองค์ประกอบ n และตัวเลข k มีนักเรียน n คนในชั้นเรียน คะแนนของนักเรียน ith คือ 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,