#include <cstdio> #include <vector> #include <map> #include <set> #include <queue> #include <algorithm> #include <string> #include <sstream> #include <cmath> using namespace std; typedef vector<int> vi; typedef long long ll; typedef vector<ll> vll; typedef pair<int,int> pii; #define mp make_pair #define REP(i,n) for (int i=0,___=(n); i<___; ++i) #define FOR(i,a,b) for (int i=(a),___=(b); i<=___; ++i) #define FORD(i,a,b) for (int i=(a),___=(b); i>=___; --i) int read() { int n; scanf("%d", &n); return n; } ll readl() { ll n; scanf("%lld", &n); return n; } char readc() { static char s[32]; scanf("%s", s); return *s; } typedef pair<int, pair<int, int> > tr; #define m3(a,b,c) make_pair((a), make_pair((b), (c))) #define st first #define nd second.first #define rd second.second bool cmp(const tr &a, const tr &b) { return a.rd < b.rd || (a.rd == b.rd && a.nd < b.nd); } bool solve() { int n = read(); int m = read(); priority_queue<tr> q; REP(i, n) { int p = read(); int k = read(); int c = read(); q.push(m3(-p, c, k - p - c)); } while (q.size() > m) { vector<tr> v; v.push_back(q.top()); q.pop(); while (!q.empty() && q.top().st == v[0].st) { v.push_back(q.top()); q.pop(); } if (v.size() <= m) { if (q.empty()) return true; int diff = v[0].st - q.top().st; REP(i, v.size()) { v[i].st -= diff; v[i].nd -= diff; if (v[i].nd > 0) { q.push(v[i]); } } } else { sort(v.begin(), v.end(), cmp); if (v[m].rd <= 0) return false; FOR(i, m, v.size()-1) { v[i].st--; v[i].rd--; q.push(v[i]); } REP(i, m) { v[i].st--; v[i].nd--; if (v[i].nd > 0) { q.push(v[i]); } } } } return true; } int main() { printf("%s\n", solve() ? "TAK" : "NIE"); 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 | #include <cstdio> #include <vector> #include <map> #include <set> #include <queue> #include <algorithm> #include <string> #include <sstream> #include <cmath> using namespace std; typedef vector<int> vi; typedef long long ll; typedef vector<ll> vll; typedef pair<int,int> pii; #define mp make_pair #define REP(i,n) for (int i=0,___=(n); i<___; ++i) #define FOR(i,a,b) for (int i=(a),___=(b); i<=___; ++i) #define FORD(i,a,b) for (int i=(a),___=(b); i>=___; --i) int read() { int n; scanf("%d", &n); return n; } ll readl() { ll n; scanf("%lld", &n); return n; } char readc() { static char s[32]; scanf("%s", s); return *s; } typedef pair<int, pair<int, int> > tr; #define m3(a,b,c) make_pair((a), make_pair((b), (c))) #define st first #define nd second.first #define rd second.second bool cmp(const tr &a, const tr &b) { return a.rd < b.rd || (a.rd == b.rd && a.nd < b.nd); } bool solve() { int n = read(); int m = read(); priority_queue<tr> q; REP(i, n) { int p = read(); int k = read(); int c = read(); q.push(m3(-p, c, k - p - c)); } while (q.size() > m) { vector<tr> v; v.push_back(q.top()); q.pop(); while (!q.empty() && q.top().st == v[0].st) { v.push_back(q.top()); q.pop(); } if (v.size() <= m) { if (q.empty()) return true; int diff = v[0].st - q.top().st; REP(i, v.size()) { v[i].st -= diff; v[i].nd -= diff; if (v[i].nd > 0) { q.push(v[i]); } } } else { sort(v.begin(), v.end(), cmp); if (v[m].rd <= 0) return false; FOR(i, m, v.size()-1) { v[i].st--; v[i].rd--; q.push(v[i]); } REP(i, m) { v[i].st--; v[i].nd--; if (v[i].nd > 0) { q.push(v[i]); } } } } return true; } int main() { printf("%s\n", solve() ? "TAK" : "NIE"); return 0; } |