#include <cstdio> #include <vector> #include <algorithm> using namespace std; double eps = 0.0000001; int n,poj,chce,ma; int prz = 0; int inf = 1000000000; bool jeden() { //printf("START\n"); vector<pair<long long, long long>> CHCE, MA; vector<bool> good; long long energia = 0; scanf("%d",&n); //if(prz == 10) printf("%d\n", n); int minDany = inf; int maxDany = -inf; int minChciany = inf; int maxChciany = -inf; for(int i=0;i<n;i++) { scanf("%d%d%d", &poj, &ma, &chce); //if(prz == 10) printf("%d %d %d\n", poj, ma, chce); minDany = min(minDany, ma); maxDany = max(maxDany, ma); minChciany = min(minChciany, chce); maxChciany = max(maxChciany, chce); CHCE.emplace_back(chce, poj); good.push_back(false); energia += poj * (ma - chce); MA.emplace_back(ma, poj); } if(energia != 0 || minDany > minChciany || maxDany < maxChciany) { return false; } sort(CHCE.begin(), CHCE.end()); sort(MA.begin(), MA.end()); int I = n - 1; long long uz = 0; long long ileMLChcemy = 0; long long ileMLMamy = 0; long long srednia = 0; long long srednia2 = 0; for(int i = n-1; i >= 0; i--) { ileMLChcemy += CHCE[i].second; srednia2 += CHCE[i].second * CHCE[i].first; while(ileMLMamy < ileMLChcemy) { long long amount = min(ileMLChcemy - ileMLMamy, MA[I].second - uz); uz += amount; ileMLMamy += amount; srednia += amount * MA[I].first; if(MA[I].second == uz) { I--; uz = 0; } } //if(srednia >= CHCE[i].first * ileMLMamy) good[i] = true;//return false; if(srednia < srednia2) return false; } I = 0; uz = 0; ileMLChcemy = 0; ileMLMamy = 0; srednia = 0; srednia2 = 0; for(int i = 0; i < n; i++) { ileMLChcemy += CHCE[i].second; srednia2 += CHCE[i].second * CHCE[i].first; while(ileMLMamy < ileMLChcemy) { long long amount = min(ileMLChcemy - ileMLMamy, MA[I].second - uz); uz += amount; ileMLMamy += amount; srednia += amount * MA[I].first; if(MA[I].second == uz) { I++; uz = 0; } } //if(srednia <= CHCE[i].first * ileMLMamy) good[i] = true;//return false; if(srednia > srednia2) return false; } for(int i = 0; i < n; i++) { //if(!good[i]) return false; } return true; } int main() { int T; scanf("%d",&T); for(int i=0;i<T;i++) { prz = i + 0; if(jeden())printf("TAK\n"); else printf("NIE\n"); } }
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 121 122 123 124 125 126 127 128 | #include <cstdio> #include <vector> #include <algorithm> using namespace std; double eps = 0.0000001; int n,poj,chce,ma; int prz = 0; int inf = 1000000000; bool jeden() { //printf("START\n"); vector<pair<long long, long long>> CHCE, MA; vector<bool> good; long long energia = 0; scanf("%d",&n); //if(prz == 10) printf("%d\n", n); int minDany = inf; int maxDany = -inf; int minChciany = inf; int maxChciany = -inf; for(int i=0;i<n;i++) { scanf("%d%d%d", &poj, &ma, &chce); //if(prz == 10) printf("%d %d %d\n", poj, ma, chce); minDany = min(minDany, ma); maxDany = max(maxDany, ma); minChciany = min(minChciany, chce); maxChciany = max(maxChciany, chce); CHCE.emplace_back(chce, poj); good.push_back(false); energia += poj * (ma - chce); MA.emplace_back(ma, poj); } if(energia != 0 || minDany > minChciany || maxDany < maxChciany) { return false; } sort(CHCE.begin(), CHCE.end()); sort(MA.begin(), MA.end()); int I = n - 1; long long uz = 0; long long ileMLChcemy = 0; long long ileMLMamy = 0; long long srednia = 0; long long srednia2 = 0; for(int i = n-1; i >= 0; i--) { ileMLChcemy += CHCE[i].second; srednia2 += CHCE[i].second * CHCE[i].first; while(ileMLMamy < ileMLChcemy) { long long amount = min(ileMLChcemy - ileMLMamy, MA[I].second - uz); uz += amount; ileMLMamy += amount; srednia += amount * MA[I].first; if(MA[I].second == uz) { I--; uz = 0; } } //if(srednia >= CHCE[i].first * ileMLMamy) good[i] = true;//return false; if(srednia < srednia2) return false; } I = 0; uz = 0; ileMLChcemy = 0; ileMLMamy = 0; srednia = 0; srednia2 = 0; for(int i = 0; i < n; i++) { ileMLChcemy += CHCE[i].second; srednia2 += CHCE[i].second * CHCE[i].first; while(ileMLMamy < ileMLChcemy) { long long amount = min(ileMLChcemy - ileMLMamy, MA[I].second - uz); uz += amount; ileMLMamy += amount; srednia += amount * MA[I].first; if(MA[I].second == uz) { I++; uz = 0; } } //if(srednia <= CHCE[i].first * ileMLMamy) good[i] = true;//return false; if(srednia > srednia2) return false; } for(int i = 0; i < n; i++) { //if(!good[i]) return false; } return true; } int main() { int T; scanf("%d",&T); for(int i=0;i<T;i++) { prz = i + 0; if(jeden())printf("TAK\n"); else printf("NIE\n"); } } |