#include<iostream> #include<set> using namespace std; int main() { ios_base::sync_with_stdio(0); cin.tie(0); set<int> s; long long n,k,j = 0,wyn = 0, a; cin >> n >> k; for (long long i = 0; i < n && j < k; i++) { cin >> a; if (s.emplace(a).second) wyn += i - (j++); } if (j < k) wyn = -1; cout << wyn; return 0; }
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 | #include<iostream> #include<set> using namespace std; int main() { ios_base::sync_with_stdio(0); cin.tie(0); set<int> s; long long n,k,j = 0,wyn = 0, a; cin >> n >> k; for (long long i = 0; i < n && j < k; i++) { cin >> a; if (s.emplace(a).second) wyn += i - (j++); } if (j < k) wyn = -1; cout << wyn; return 0; } |