#include <iostream>
#include <vector>
#include <algorithm>
#include <math.h>
using namespace std;
void sortVector(vector<pair<int, int>> &vectorToSort)
{
sort(vectorToSort.begin(), vectorToSort.end(),
[](const pair<int, int> & a, const pair<int, int> & b) -> bool
{
return a.second > b.second;
});
}
void oneTest()
{
int numberOfBajtoniatek, capacity, inputTemp, outputTemp;
long long int currentTempInput = 0, currentTempOutput = 0;
long long int currentCapacityInput = 0, currentCapacityOutput = 0;
vector<pair<int, int>> inputData;
vector<pair<int, int>> outputData;
cin>> numberOfBajtoniatek;
for(int i = 0; i < numberOfBajtoniatek; i++)
{
cin>>capacity>>inputTemp>>outputTemp;
inputData.emplace_back(capacity, inputTemp);
outputData.emplace_back(capacity, outputTemp);
}
sortVector(inputData);
sortVector(outputData);
int indexInInput = 0;
for(int i = 0; i < outputData.size(); i++)
{
while(currentCapacityInput + inputData[indexInInput].first < currentCapacityOutput)
{
currentTempInput = (currentTempInput + inputData[indexInInput].first*inputData[indexInInput].second);
currentCapacityInput += inputData[indexInInput].first;
indexInInput++;
}
if(currentCapacityInput < currentCapacityOutput)
{
long long int howMuchToSubtract = currentCapacityOutput - currentCapacityInput;
inputData[indexInInput].first -= howMuchToSubtract;
currentTempInput = (currentTempInput + howMuchToSubtract*inputData[indexInInput].second);
currentCapacityInput += howMuchToSubtract;
}
// cout<<"current temp input "<< currentTempInput << " currentTempOutput"<<currentTempOutput << " currentCapacityOutput: "<< currentCapacityOutput << " "<< currentCapacityInput<<endl;
if(currentTempInput < currentTempOutput)
{
cout<<"NIE"<<endl; // <<currentTempInput - currentTempOutput<<endl;
return;
}
currentTempOutput = currentTempOutput + outputData[i].first*outputData[i].second;
currentCapacityOutput = currentCapacityOutput + outputData[i].first;
}
while(currentCapacityInput < currentCapacityOutput)
{
currentTempInput = currentTempInput + inputData[indexInInput].first*inputData[indexInInput].second;
currentCapacityInput += inputData[indexInInput].first;
indexInInput++;
}
if(currentTempInput == currentTempOutput)
{
cout<<"TAK"<<endl;
return;
}
cout<<"NIE"<<endl;
}
int main() {
ios_base::sync_with_stdio(false);
int numberOfTests;
cin>>numberOfTests;
for(int i = 0; i < numberOfTests; i++)
{
oneTest();
}
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 | #include <iostream> #include <vector> #include <algorithm> #include <math.h> using namespace std; void sortVector(vector<pair<int, int>> &vectorToSort) { sort(vectorToSort.begin(), vectorToSort.end(), [](const pair<int, int> & a, const pair<int, int> & b) -> bool { return a.second > b.second; }); } void oneTest() { int numberOfBajtoniatek, capacity, inputTemp, outputTemp; long long int currentTempInput = 0, currentTempOutput = 0; long long int currentCapacityInput = 0, currentCapacityOutput = 0; vector<pair<int, int>> inputData; vector<pair<int, int>> outputData; cin>> numberOfBajtoniatek; for(int i = 0; i < numberOfBajtoniatek; i++) { cin>>capacity>>inputTemp>>outputTemp; inputData.emplace_back(capacity, inputTemp); outputData.emplace_back(capacity, outputTemp); } sortVector(inputData); sortVector(outputData); int indexInInput = 0; for(int i = 0; i < outputData.size(); i++) { while(currentCapacityInput + inputData[indexInInput].first < currentCapacityOutput) { currentTempInput = (currentTempInput + inputData[indexInInput].first*inputData[indexInInput].second); currentCapacityInput += inputData[indexInInput].first; indexInInput++; } if(currentCapacityInput < currentCapacityOutput) { long long int howMuchToSubtract = currentCapacityOutput - currentCapacityInput; inputData[indexInInput].first -= howMuchToSubtract; currentTempInput = (currentTempInput + howMuchToSubtract*inputData[indexInInput].second); currentCapacityInput += howMuchToSubtract; } // cout<<"current temp input "<< currentTempInput << " currentTempOutput"<<currentTempOutput << " currentCapacityOutput: "<< currentCapacityOutput << " "<< currentCapacityInput<<endl; if(currentTempInput < currentTempOutput) { cout<<"NIE"<<endl; // <<currentTempInput - currentTempOutput<<endl; return; } currentTempOutput = currentTempOutput + outputData[i].first*outputData[i].second; currentCapacityOutput = currentCapacityOutput + outputData[i].first; } while(currentCapacityInput < currentCapacityOutput) { currentTempInput = currentTempInput + inputData[indexInInput].first*inputData[indexInInput].second; currentCapacityInput += inputData[indexInInput].first; indexInInput++; } if(currentTempInput == currentTempOutput) { cout<<"TAK"<<endl; return; } cout<<"NIE"<<endl; } int main() { ios_base::sync_with_stdio(false); int numberOfTests; cin>>numberOfTests; for(int i = 0; i < numberOfTests; i++) { oneTest(); } return 0; } |
English