Algorithm
Olya likes milk very much. She drinks k cartons of milk each day if she has at least k and drinks all of them if she doesn't. But there's an issue — expiration dates. Each carton has a date after which you can't drink it (you still can drink it exactly at the date written on the carton). Due to this, if Olya's fridge contains a carton past its expiry date, she throws it away.
Olya hates throwing out cartons, so when she drinks a carton, she chooses the one which expires the fastest. It's easy to understand that this strategy minimizes the amount of cartons thrown out and lets her avoid it if it's even possible.
![](https://espresso.codeforces.com/ab13b6fe3c5d4854e97d167ce78ffe3d8d35a95e.png)
The main issue Olya has is the one of buying new cartons. Currently, there are n cartons of milk in Olya's fridge, for each one an expiration date is known (how soon does it expire, measured in days). In the shop that Olya visited there are m cartons, and the expiration date is known for each of those cartons as well.
Find the maximum number of cartons Olya can buy so that she wouldn't have to throw away any cartons. Assume that Olya drank no cartons today.
In the first line there are three integers n, m, k (1 ≤ n, m ≤ 106, 1 ≤ k ≤ n + m) — the amount of cartons in Olya's fridge, the amount of cartons in the shop and the number of cartons Olya drinks each day.
In the second line there are n integers f1, f2, ..., fn (0 ≤ fi ≤ 107) — expiration dates of the cartons in Olya's fridge. The expiration date is expressed by the number of days the drinking of this carton can be delayed. For example, a 0 expiration date means it must be drunk today, 1 — no later than tomorrow, etc.
In the third line there are m integers s1, s2, ..., sm (0 ≤ si ≤ 107) — expiration dates of the cartons in the shop in a similar format.
If there's no way for Olya to drink the cartons she already has in her fridge, print -1.
Otherwise, in the first line print the maximum number x of cartons which Olya can buy so that she wouldn't have to throw a carton away. The next line should contain exactly x integers — the numbers of the cartons that should be bought (cartons are numbered in an order in which they are written in the input, starting with 1). Numbers should not repeat, but can be in arbitrary order. If there are multiple correct answers, print any of them.
3 6 2
1 0 1
2 0 2 0 0 2
3
1 2 3
3 1 2
0 0 0
1
-1
2 1 2
0 1
0
1
1
In the first example k = 2 and Olya has three cartons with expiry dates 0, 1 and 1 (they expire today, tomorrow and tomorrow), and the shop has 3 cartons with expiry date 0 and 3 cartons with expiry date 2. Olya can buy three cartons, for example, one with the expiry date 0 and two with expiry date 2.
In the second example all three cartons Olya owns expire today and it means she would have to throw packets away regardless of whether she buys an extra one or not.
In the third example Olya would drink k = 2 cartons today (one she alreay has in her fridge and one from the shop) and the remaining one
Code Examples
#1 Code Example with C++ Programming
Code -
C++ Programming
#include <bits/stdc++.h>
using namespace std;
int const N = 1e6 + 1, M = 1e7 + 1;
int n, m, k, f[N], fr[M];
pair<int, int> s[N];
vector<int> all;
bool can(int mid) {
memset(fr, 0, sizeof fr);
for(int i = 0; i < n; ++i)
++fr[f[i]];
for(int i = 0; i < mid; ++i)
++fr[s[i].first];
int day = 0, c = 0;
for(int i = 0; i < (int)all.size(); ++i) {
while(fr[all[i]]) {
if(c == k) {
c = 0;
++day;
}
if(day > all[i])
return false;
++c;
--fr[all[i]];
}
}
return true;
}
int main() {
scanf("%d %d %d", &n, &m, &k);
for(int i = 0; i < n; ++i) {
scanf("%d", f + i);
all.push_back(f[i]);
}
for(int i = 0; i < m; ++i) {
scanf("%d", &s[i].first);
all.push_back(s[i].first);
s[i].second = i + 1;
}
sort(s, s + m);
reverse(s, s + m);
sort(all.begin(), all.end());
all.resize(unique(all.begin(), all.end()) - all.begin());
int l = 0, r = m, mid, res = -1;
while(l <= r) {
mid = (l + r) >> 1;
if(can(mid)) {
l = mid + 1;
res = mid;
} else
r = mid - 1;
}
printf("%d\n", res);
for(int i = 0; i < res; ++i)
printf("%d ", s[i].second);
return 0;
}
Copy The Code &
Try With Live Editor
Input
1 0 1
2 0 2 0 0 2
Output
1 2 3
Demonstration
Codeforces Solution-Cartons of milk-Solution in C, C++, Java, Python