/* Author: Dominik Wójt */
/* Problem: Bohater, Potyczki Algorytmiczne 2014 */
#include <iostream>
#include <cstdlib>
#include <vector>
#include <algorithm>
#define MY_ASSERT(condition) \
do \
{ \
if(!(condition)) \
{ \
std::cerr << "assert failed: " << #condition << std::endl; \
std::abort(); \
} \
} \
while(false)
//------------------------------------------------------------------------------
typedef long int lint;
typedef long int llint;
//------------------------------------------------------------------------------
template<typename T>
T checked_read(const T min, const T max)
{
T n;
std::cin >> n;
MY_ASSERT(std::cin);
MY_ASSERT(min<=n && n<=max);
return n;
}
//------------------------------------------------------------------------------
struct Monster
{
lint cost;
lint reward;
lint index;
};
//------------------------------------------------------------------------------
lint gain(Monster monster)
{
return monster.reward - monster.cost;
}
//------------------------------------------------------------------------------
lint less_by_cost(Monster a, Monster b)
{
return a.cost < b.cost;
}
//------------------------------------------------------------------------------
lint greater_by_reward(Monster a, Monster b)
{
return a.reward > b.reward;
}
//------------------------------------------------------------------------------
void make_optimal_order(std::vector<Monster> &monsters)
{
typedef std::vector<Monster>::iterator iterator;
iterator left = monsters.begin();
iterator right = monsters.end()-1;
while(left<right)
{
while(left<right && gain(*left)>=0)
left++;
while(left<right && gain(*right)<0)
right--;
while(left<right && gain(*left)<0 && gain(*right)>=0)
{
std::swap(*left,*right);
left++;
right--;
}
}
if(gain(*left)>=0)
left++;
MY_ASSERT(left>=monsters.end() || gain(*left)<0);
MY_ASSERT(left-1<monsters.begin() || gain(*(left-1))>=0);
//at this point everything in range begin to left should have gain>=0
std::sort(monsters.begin(), left, less_by_cost);
std::sort(left, monsters.end(), greater_by_reward);
}
//------------------------------------------------------------------------------
bool check_game_success( const lint start_health,
const std::vector<Monster> &monsters)
{
llint health = start_health;
typedef std::vector<Monster>::const_iterator iterator;
for(iterator monster=monsters.begin(); monster<monsters.end(); ++monster)
{
if(monster->cost >= health)
return false;
health += gain(*monster);
}
return true;
}
//------------------------------------------------------------------------------
int main()
{
const lint n = checked_read<lint>( 1, 100000 );
const lint z = checked_read<lint>( 1, 100000 );
std::vector<Monster> monsters( n );
for(int i=0; i<n; i++)
{
monsters[i].cost = checked_read<lint>( 0, 100000 );
monsters[i].reward = checked_read<lint>( 0, 100000 );
monsters[i].index = i;
}
make_optimal_order(monsters);
if(check_game_success(z, monsters))
{
std::cout << "TAK\n";
for(int i=0; i<n-1; i++)
{
std::cout << monsters[i].index+1 << " ";
}
std::cout << monsters.back().index+1 << "\n";
}
else
{
std::cout << "NIE\n";
}
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 128 | /* Author: Dominik Wójt */ /* Problem: Bohater, Potyczki Algorytmiczne 2014 */ #include <iostream> #include <cstdlib> #include <vector> #include <algorithm> #define MY_ASSERT(condition) \ do \ { \ if(!(condition)) \ { \ std::cerr << "assert failed: " << #condition << std::endl; \ std::abort(); \ } \ } \ while(false) //------------------------------------------------------------------------------ typedef long int lint; typedef long int llint; //------------------------------------------------------------------------------ template<typename T> T checked_read(const T min, const T max) { T n; std::cin >> n; MY_ASSERT(std::cin); MY_ASSERT(min<=n && n<=max); return n; } //------------------------------------------------------------------------------ struct Monster { lint cost; lint reward; lint index; }; //------------------------------------------------------------------------------ lint gain(Monster monster) { return monster.reward - monster.cost; } //------------------------------------------------------------------------------ lint less_by_cost(Monster a, Monster b) { return a.cost < b.cost; } //------------------------------------------------------------------------------ lint greater_by_reward(Monster a, Monster b) { return a.reward > b.reward; } //------------------------------------------------------------------------------ void make_optimal_order(std::vector<Monster> &monsters) { typedef std::vector<Monster>::iterator iterator; iterator left = monsters.begin(); iterator right = monsters.end()-1; while(left<right) { while(left<right && gain(*left)>=0) left++; while(left<right && gain(*right)<0) right--; while(left<right && gain(*left)<0 && gain(*right)>=0) { std::swap(*left,*right); left++; right--; } } if(gain(*left)>=0) left++; MY_ASSERT(left>=monsters.end() || gain(*left)<0); MY_ASSERT(left-1<monsters.begin() || gain(*(left-1))>=0); //at this point everything in range begin to left should have gain>=0 std::sort(monsters.begin(), left, less_by_cost); std::sort(left, monsters.end(), greater_by_reward); } //------------------------------------------------------------------------------ bool check_game_success( const lint start_health, const std::vector<Monster> &monsters) { llint health = start_health; typedef std::vector<Monster>::const_iterator iterator; for(iterator monster=monsters.begin(); monster<monsters.end(); ++monster) { if(monster->cost >= health) return false; health += gain(*monster); } return true; } //------------------------------------------------------------------------------ int main() { const lint n = checked_read<lint>( 1, 100000 ); const lint z = checked_read<lint>( 1, 100000 ); std::vector<Monster> monsters( n ); for(int i=0; i<n; i++) { monsters[i].cost = checked_read<lint>( 0, 100000 ); monsters[i].reward = checked_read<lint>( 0, 100000 ); monsters[i].index = i; } make_optimal_order(monsters); if(check_game_success(z, monsters)) { std::cout << "TAK\n"; for(int i=0; i<n-1; i++) { std::cout << monsters[i].index+1 << " "; } std::cout << monsters.back().index+1 << "\n"; } else { std::cout << "NIE\n"; } return 0; } |
English