#include <algorithm>
#include <iostream>
#include <vector>
using namespace std;
namespace {
struct Task {
int start;
int deadline;
int time;
Task(int s, int d, int t): start{s}, deadline{d}, time{t}
{
}
};
using ll = long long;
bool possible(vector<Task>&& tasks, int m)
{
sort(tasks.begin(), tasks.end(), [](Task const& lhs, Task const& rhs) {
return lhs.start < rhs.start;
});
auto it = tasks.begin();
int now = it->start;
vector<Task*> waiting;
while (it != tasks.end() || !waiting.empty()) {
if (waiting.empty()) now = it->start;
while (it != tasks.end() && it->start == now) {
waiting.emplace_back(&*it);
++it;
}
for (Task* t: waiting) {
if (t->deadline <= now) return false;
}
int w = waiting.size();
auto k = min(w, m);
if (w > m) {
nth_element(waiting.begin(), waiting.begin() + k, waiting.end(), [&](Task* a, Task* b) {
return ll{a->time} * (b->deadline - now) > ll{b->time} * (a->deadline - now);
});
}
for (int i = 0; i < k; ++i) {
--waiting[i]->time;
}
waiting.erase(remove_if(waiting.begin(), waiting.end(), [&](Task* t) { return t->time == 0; }), waiting.end());
++now;
}
return true;
}
}
int main()
{
iostream::sync_with_stdio(false);
cin.tie(nullptr);
int n, m;
cin >> n >> m;
vector<Task> tasks;
tasks.reserve(n);
for (int i = 0; i < n; ++i) {
int p, k, c;
cin >> p >> k >> c;
tasks.emplace_back(p, k, c);
}
cout << (possible(move(tasks), m) ? "TAK" : "NIE") << '\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 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 | #include <algorithm> #include <iostream> #include <vector> using namespace std; namespace { struct Task { int start; int deadline; int time; Task(int s, int d, int t): start{s}, deadline{d}, time{t} { } }; using ll = long long; bool possible(vector<Task>&& tasks, int m) { sort(tasks.begin(), tasks.end(), [](Task const& lhs, Task const& rhs) { return lhs.start < rhs.start; }); auto it = tasks.begin(); int now = it->start; vector<Task*> waiting; while (it != tasks.end() || !waiting.empty()) { if (waiting.empty()) now = it->start; while (it != tasks.end() && it->start == now) { waiting.emplace_back(&*it); ++it; } for (Task* t: waiting) { if (t->deadline <= now) return false; } int w = waiting.size(); auto k = min(w, m); if (w > m) { nth_element(waiting.begin(), waiting.begin() + k, waiting.end(), [&](Task* a, Task* b) { return ll{a->time} * (b->deadline - now) > ll{b->time} * (a->deadline - now); }); } for (int i = 0; i < k; ++i) { --waiting[i]->time; } waiting.erase(remove_if(waiting.begin(), waiting.end(), [&](Task* t) { return t->time == 0; }), waiting.end()); ++now; } return true; } } int main() { iostream::sync_with_stdio(false); cin.tie(nullptr); int n, m; cin >> n >> m; vector<Task> tasks; tasks.reserve(n); for (int i = 0; i < n; ++i) { int p, k, c; cin >> p >> k >> c; tasks.emplace_back(p, k, c); } cout << (possible(move(tasks), m) ? "TAK" : "NIE") << '\n'; return 0; } |
English