#include <cstdio> #include <tuple> #include <vector> #include <algorithm> #include <set> #include <functional> void empty(int n_args, ...) { } // #define _DEBUG 1; #ifdef _DEBUG #define dprintf printf #else #define dprintf empty; #endif struct Task { int p, k, c; Task(int pp, int kk, int cc): p(pp), k(kk), c(cc) {} }; struct cmp { bool operator() (const Task& a, const Task& b) const{ if(a.p != b.p) { return a.p < b.p; } return a.c < b.c; } }; struct SuperMap { std::set<Task, cmp> pool; void init(int procs, int minp, int maxk) { for(int i = 0; i < procs; ++i) { pool.insert(Task(minp, maxk, i)); } } void print_pool() { for(auto it = pool.begin(); it != pool.end(); ++it) { dprintf("%d-%d proc %d, ", it->p, it->k, it->c); } dprintf("\n"); } bool reserve(Task& task) { print_pool(); dprintf("reserve task %d %d %d\n", task.p, task.k, task.c); int time_left = task.c, pos = task.p; while(time_left > 0) { dprintf("tl=%d pos=%d\n", time_left, pos); auto it = pool.lower_bound(Task(pos, 0, -1)); Task range = *it; if(it == pool.end()) { dprintf("end of pools :(\n"); return false; } if(time_left > range.k - pos) { dprintf("not enough, take from %d to %d\n", pos, range.k); time_left -= range.k - pos; pos = range.k; pool.erase(it); if(range.p < pos) { pool.insert(Task(range.p, pos, range.c)); } } else { dprintf("enough, split? on %d\n", range.p + time_left); pool.erase(it); if(range.p < pos) { pool.insert(Task(range.p, pos, range.c)); } if(range.p + time_left < range.k) { pool.insert(Task(range.p + time_left, range.k, range.c)); } time_left = 0; } } dprintf("\n"); return true; } }; void solve() { int n, m; int p, k, c; int minp=1000000, maxk=0; std::vector<Task> tasks; scanf("%d %d", &n, &m); for(int i = 0; i < n; ++i) { scanf("%d %d %d", &p, &k, &c); minp = std::min(minp, p); maxk = std::max(maxk, k); tasks.push_back(Task(p, k, c)); } // dprintf("minp %d maxk %d\n", minp, maxk); SuperMap sm; sm.init(m, minp, maxk); std::sort(tasks.begin(), tasks.end(), [](const Task &a, const Task &b) { return a.k < b.k; }); // for(auto i = tasks.begin(); i != tasks.end(); ++i) // dprintf("%d %d %d\n", i->p, i->k, i->c); for(auto i = tasks.begin(); i != tasks.end(); ++i) { Task task = *i; if(!sm.reserve(task)) { printf("NIE\n"); return; } } sm.print_pool(); printf("TAK\n"); } int main() { solve(); 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 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 | #include <cstdio> #include <tuple> #include <vector> #include <algorithm> #include <set> #include <functional> void empty(int n_args, ...) { } // #define _DEBUG 1; #ifdef _DEBUG #define dprintf printf #else #define dprintf empty; #endif struct Task { int p, k, c; Task(int pp, int kk, int cc): p(pp), k(kk), c(cc) {} }; struct cmp { bool operator() (const Task& a, const Task& b) const{ if(a.p != b.p) { return a.p < b.p; } return a.c < b.c; } }; struct SuperMap { std::set<Task, cmp> pool; void init(int procs, int minp, int maxk) { for(int i = 0; i < procs; ++i) { pool.insert(Task(minp, maxk, i)); } } void print_pool() { for(auto it = pool.begin(); it != pool.end(); ++it) { dprintf("%d-%d proc %d, ", it->p, it->k, it->c); } dprintf("\n"); } bool reserve(Task& task) { print_pool(); dprintf("reserve task %d %d %d\n", task.p, task.k, task.c); int time_left = task.c, pos = task.p; while(time_left > 0) { dprintf("tl=%d pos=%d\n", time_left, pos); auto it = pool.lower_bound(Task(pos, 0, -1)); Task range = *it; if(it == pool.end()) { dprintf("end of pools :(\n"); return false; } if(time_left > range.k - pos) { dprintf("not enough, take from %d to %d\n", pos, range.k); time_left -= range.k - pos; pos = range.k; pool.erase(it); if(range.p < pos) { pool.insert(Task(range.p, pos, range.c)); } } else { dprintf("enough, split? on %d\n", range.p + time_left); pool.erase(it); if(range.p < pos) { pool.insert(Task(range.p, pos, range.c)); } if(range.p + time_left < range.k) { pool.insert(Task(range.p + time_left, range.k, range.c)); } time_left = 0; } } dprintf("\n"); return true; } }; void solve() { int n, m; int p, k, c; int minp=1000000, maxk=0; std::vector<Task> tasks; scanf("%d %d", &n, &m); for(int i = 0; i < n; ++i) { scanf("%d %d %d", &p, &k, &c); minp = std::min(minp, p); maxk = std::max(maxk, k); tasks.push_back(Task(p, k, c)); } // dprintf("minp %d maxk %d\n", minp, maxk); SuperMap sm; sm.init(m, minp, maxk); std::sort(tasks.begin(), tasks.end(), [](const Task &a, const Task &b) { return a.k < b.k; }); // for(auto i = tasks.begin(); i != tasks.end(); ++i) // dprintf("%d %d %d\n", i->p, i->k, i->c); for(auto i = tasks.begin(); i != tasks.end(); ++i) { Task task = *i; if(!sm.reserve(task)) { printf("NIE\n"); return; } } sm.print_pool(); printf("TAK\n"); } int main() { solve(); return 0; } |