#include <iostream>
#include <set>
using namespace std;
int main()
{
int n, requiredUnique;
int totalTime = 0;
set<int> uniqueNums;
cin >> n >> requiredUnique;
int* nums = new int[n];
for (int i = 0; i < n; i++) {
cin >> nums[i];
}
int replaceIndex = 1, replaceIndexNum = nums[1];
uniqueNums.insert(nums[0]);
for (int i = 1; i < n; i++) {
if (uniqueNums.size() >= requiredUnique) {
break;
}
auto position = uniqueNums.find(nums[i]);
//cerr << "replaceIndex: " << replaceIndex << "\t";
if (position == uniqueNums.end()) {
//cerr << "Not in set -> " << nums[i] << "\t";
totalTime += i - replaceIndex;
replaceIndex++;
replaceIndexNum = nums[replaceIndex];
uniqueNums.insert(nums[i]);
}
//cerr << "size: " << uniqueNums.size() << "\t totalTime: " << totalTime << "\n";
}
if (uniqueNums.size() < requiredUnique) {
cout << -1 << "\n";
}
else {
cout << totalTime << "\n";
}
return 0;
}
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 | #include <iostream> #include <set> using namespace std; int main() { int n, requiredUnique; int totalTime = 0; set<int> uniqueNums; cin >> n >> requiredUnique; int* nums = new int[n]; for (int i = 0; i < n; i++) { cin >> nums[i]; } int replaceIndex = 1, replaceIndexNum = nums[1]; uniqueNums.insert(nums[0]); for (int i = 1; i < n; i++) { if (uniqueNums.size() >= requiredUnique) { break; } auto position = uniqueNums.find(nums[i]); //cerr << "replaceIndex: " << replaceIndex << "\t"; if (position == uniqueNums.end()) { //cerr << "Not in set -> " << nums[i] << "\t"; totalTime += i - replaceIndex; replaceIndex++; replaceIndexNum = nums[replaceIndex]; uniqueNums.insert(nums[i]); } //cerr << "size: " << uniqueNums.size() << "\t totalTime: " << totalTime << "\n"; } if (uniqueNums.size() < requiredUnique) { cout << -1 << "\n"; } else { cout << totalTime << "\n"; } return 0; } |
English