#include <iostream>
#include <vector>
#include <numeric>
#include <algorithm>
//Kisielinski Damian
using namespace std;
struct Person
{
double litre;
double currTemp;
double tarrTemp;
};
vector<int> sortIndexesCurr(Person * v, int len) {
vector<int> idx(len);
iota(idx.begin(), idx.end(), 0);
sort(idx.begin(), idx.end(),
[v](int i1, int i2) {return v[i1].currTemp > v[i2].currTemp;});
return idx;
}
vector<int> sortIndexesTarr(Person * v, int len) {
vector<int> idx(len);
iota(idx.begin(), idx.end(), 0);
sort(idx.begin(), idx.end(),
[v](int i1, int i2) {return v[i1].tarrTemp > v[i2].tarrTemp;});
return idx;
}
bool areSame(double a, double b)
{
return (a - b) > 0 ? (a - b) < 0.000000001 : (b-a) < 0.000000001;
}
int main(int argc, char** argv) {
ios_base::sync_with_stdio(0);
int n, t;
cin>>t;
for (int i = 0; i < t; i++) {
//cout<<"Zestaw: "<<i<<endl;
cin>>n;
Person tab[n];
double currSum=0, tarrSum=0;
for (int j = 0; j < n; j++) {
cin>>tab[j].litre>>tab[j].currTemp>>tab[j].tarrTemp;
currSum+=(long)tab[j].currTemp;
tarrSum+=(long)tab[j].tarrTemp;
}
if (currSum != tarrSum) {
cout<<"NIE"<<endl;
continue;
}
if (n == 1) {
cout<<"TAK"<<endl;
continue;
}
vector<int> byTarr = sortIndexesTarr(tab, n);
vector<int> byCurr = sortIndexesCurr(tab, n);
double litrChild=tab[byCurr[1]].litre, tempVol=tab[byCurr[0]].litre, tempTemp=tab[byCurr[0]].currTemp;
int indChild=1;
bool fail = false;
for (int j = 0; j < n; j++) {
//cout<<tab[byCurr[j]].currTemp<<" "<<tab[byTarr[j]].tarrTemp<<endl;
if (fail) {
break;
}
if (tab[byTarr[j]].tarrTemp > tempTemp) {
fail = true;
break;
}
while ((!areSame(tab[byTarr[j]].tarrTemp, tempTemp) || !tab[byTarr[j]].litre <= tempVol) && indChild < n) {
if (areSame(tab[byTarr[j]].tarrTemp, tempTemp)) { // licznik rowny zero
if (areSame(tab[byCurr[indChild]].currTemp, tempTemp)) { // licznik równy zero
tempVol += litrChild;
indChild++;
if (indChild < n) {
litrChild = tab[byCurr[indChild]].litre;
} else {
litrChild = 0;
}
continue;
} else {
fail = true;
break;
}
}
if (tab[byTarr[j]].tarrTemp > tab[byCurr[indChild]].currTemp) {
fail = true;
break;
}
double a = tempVol * (tempTemp - tab[byTarr[j]].tarrTemp) / (tab[byTarr[j]].tarrTemp - tab[byCurr[indChild]].currTemp);
if (a < litrChild) {
litrChild -= a;
tempVol += a;
tempTemp = tab[byTarr[j]].tarrTemp;
} else {
tempTemp = (tab[byCurr[indChild]].currTemp * litrChild + tempTemp * tempVol)/(litrChild + tempVol);
tempVol += litrChild;
indChild++;
if (indChild < n) {
litrChild = tab[byCurr[indChild]].litre;
} else {
litrChild = 0;
}
}
}
tempVol -= tab[byTarr[j]].litre;
}
if (fail) {
cout<<"NIE"<<endl;
} else {
cout<<"TAK"<<endl;
}
}
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 121 122 123 124 125 126 127 | #include <iostream> #include <vector> #include <numeric> #include <algorithm> //Kisielinski Damian using namespace std; struct Person { double litre; double currTemp; double tarrTemp; }; vector<int> sortIndexesCurr(Person * v, int len) { vector<int> idx(len); iota(idx.begin(), idx.end(), 0); sort(idx.begin(), idx.end(), [v](int i1, int i2) {return v[i1].currTemp > v[i2].currTemp;}); return idx; } vector<int> sortIndexesTarr(Person * v, int len) { vector<int> idx(len); iota(idx.begin(), idx.end(), 0); sort(idx.begin(), idx.end(), [v](int i1, int i2) {return v[i1].tarrTemp > v[i2].tarrTemp;}); return idx; } bool areSame(double a, double b) { return (a - b) > 0 ? (a - b) < 0.000000001 : (b-a) < 0.000000001; } int main(int argc, char** argv) { ios_base::sync_with_stdio(0); int n, t; cin>>t; for (int i = 0; i < t; i++) { //cout<<"Zestaw: "<<i<<endl; cin>>n; Person tab[n]; double currSum=0, tarrSum=0; for (int j = 0; j < n; j++) { cin>>tab[j].litre>>tab[j].currTemp>>tab[j].tarrTemp; currSum+=(long)tab[j].currTemp; tarrSum+=(long)tab[j].tarrTemp; } if (currSum != tarrSum) { cout<<"NIE"<<endl; continue; } if (n == 1) { cout<<"TAK"<<endl; continue; } vector<int> byTarr = sortIndexesTarr(tab, n); vector<int> byCurr = sortIndexesCurr(tab, n); double litrChild=tab[byCurr[1]].litre, tempVol=tab[byCurr[0]].litre, tempTemp=tab[byCurr[0]].currTemp; int indChild=1; bool fail = false; for (int j = 0; j < n; j++) { //cout<<tab[byCurr[j]].currTemp<<" "<<tab[byTarr[j]].tarrTemp<<endl; if (fail) { break; } if (tab[byTarr[j]].tarrTemp > tempTemp) { fail = true; break; } while ((!areSame(tab[byTarr[j]].tarrTemp, tempTemp) || !tab[byTarr[j]].litre <= tempVol) && indChild < n) { if (areSame(tab[byTarr[j]].tarrTemp, tempTemp)) { // licznik rowny zero if (areSame(tab[byCurr[indChild]].currTemp, tempTemp)) { // licznik równy zero tempVol += litrChild; indChild++; if (indChild < n) { litrChild = tab[byCurr[indChild]].litre; } else { litrChild = 0; } continue; } else { fail = true; break; } } if (tab[byTarr[j]].tarrTemp > tab[byCurr[indChild]].currTemp) { fail = true; break; } double a = tempVol * (tempTemp - tab[byTarr[j]].tarrTemp) / (tab[byTarr[j]].tarrTemp - tab[byCurr[indChild]].currTemp); if (a < litrChild) { litrChild -= a; tempVol += a; tempTemp = tab[byTarr[j]].tarrTemp; } else { tempTemp = (tab[byCurr[indChild]].currTemp * litrChild + tempTemp * tempVol)/(litrChild + tempVol); tempVol += litrChild; indChild++; if (indChild < n) { litrChild = tab[byCurr[indChild]].litre; } else { litrChild = 0; } } } tempVol -= tab[byTarr[j]].litre; } if (fail) { cout<<"NIE"<<endl; } else { cout<<"TAK"<<endl; } } return 0; } |
English