#include <set>
#include <stdio.h>
#include <vector>
int main()
{
int n, k, bottle;
scanf("%d %d", &n, &k);
std::vector<int> bottles;
std::set<int> uniqueBottles;
for (int i = 0; i < n; ++i)
{
scanf("%d", &bottle);
bottles.push_back(bottle);
uniqueBottles.insert(bottle);
}
if (static_cast<int>(uniqueBottles.size()) < k)
{
printf("-1\n");
}
else
{
std::vector<int> bottlesToRemove;
std::vector<int> bottlesToUse;
std::set<int> usedBottles;
for (int i = 0; i < n; ++i)
{
const auto bottle = bottles[i];
if (i < k)
{
if (usedBottles.count(bottle))
{
bottlesToRemove.push_back(i);
}
else
{
usedBottles.insert(bottle);
}
}
else if (!usedBottles.count(bottle))
{
bottlesToUse.push_back(i);
usedBottles.insert(bottle);
if (bottlesToUse.size() == bottlesToRemove.size())
{
break;
}
}
}
int moves{0};
for (int i = static_cast<int>(bottlesToRemove.size()) - 1; i >= 0; --i)
{
moves += bottlesToUse[i] - bottlesToRemove[i];
}
printf("%d\n", moves);
}
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 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 | #include <set> #include <stdio.h> #include <vector> int main() { int n, k, bottle; scanf("%d %d", &n, &k); std::vector<int> bottles; std::set<int> uniqueBottles; for (int i = 0; i < n; ++i) { scanf("%d", &bottle); bottles.push_back(bottle); uniqueBottles.insert(bottle); } if (static_cast<int>(uniqueBottles.size()) < k) { printf("-1\n"); } else { std::vector<int> bottlesToRemove; std::vector<int> bottlesToUse; std::set<int> usedBottles; for (int i = 0; i < n; ++i) { const auto bottle = bottles[i]; if (i < k) { if (usedBottles.count(bottle)) { bottlesToRemove.push_back(i); } else { usedBottles.insert(bottle); } } else if (!usedBottles.count(bottle)) { bottlesToUse.push_back(i); usedBottles.insert(bottle); if (bottlesToUse.size() == bottlesToRemove.size()) { break; } } } int moves{0}; for (int i = static_cast<int>(bottlesToRemove.size()) - 1; i >= 0; --i) { moves += bottlesToUse[i] - bottlesToRemove[i]; } printf("%d\n", moves); } return 0; } |
English