#include <stdio.h>
#include <set>
#include <queue>
#include <stdint.h>
using namespace std;
inline int get_int()
{
int ret = 0;
int c = getchar_unlocked();
while(c<'0' || c>'9') c = getchar_unlocked();
while(c>='0' && c<='9')
{
ret = (ret<<3) + (ret<<1) + (c-'0');
c = getchar_unlocked();
}
return ret;
}
typedef pair<int, int> ii;
typedef pair<ii, int> item;
struct Monster
{
int d;
int a;
int r;
};
Monster monsters[100000+1];
bool fight(priority_queue<pair<int, int> > & Q, int64_t & Z, queue<int> & R)
{
while(!Q.empty() && Z > 0)
{
int i = Q.top().second;
if(Z > (int64_t)monsters[i].d)
{
Z += (int64_t)monsters[i].r;
Q.pop();
if(Z > 0)
{
R.push(i);
}
}
else
{
return false;
}
}
return true;
}
int main()
{
int N = get_int();
int64_t Z = (int64_t)get_int();
priority_queue<pair<int, int> > Qp;
priority_queue<pair<int, int> > Qm;
for(int i = 1; i <= N; i++)
{
monsters[i].d = get_int();
monsters[i].a = get_int();
monsters[i].r = monsters[i].a - monsters[i].d;
if(monsters[i].r > 0)
{
Qp.push(pair<int, int>(-monsters[i].d, i));
}
else
{
Qm.push(pair<int, int>(monsters[i].d, i));
}
}
queue<int> R;
bool ret = fight(Qp, Z, R);
if(ret)
{
fight(Qm, Z, R);
}
if(Z > 0 && R.size() == N)
{
printf("TAK\n");
while(!R.empty())
{
printf("%d ", R.front());
R.pop();
}
printf("\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 | #include <stdio.h> #include <set> #include <queue> #include <stdint.h> using namespace std; inline int get_int() { int ret = 0; int c = getchar_unlocked(); while(c<'0' || c>'9') c = getchar_unlocked(); while(c>='0' && c<='9') { ret = (ret<<3) + (ret<<1) + (c-'0'); c = getchar_unlocked(); } return ret; } typedef pair<int, int> ii; typedef pair<ii, int> item; struct Monster { int d; int a; int r; }; Monster monsters[100000+1]; bool fight(priority_queue<pair<int, int> > & Q, int64_t & Z, queue<int> & R) { while(!Q.empty() && Z > 0) { int i = Q.top().second; if(Z > (int64_t)monsters[i].d) { Z += (int64_t)monsters[i].r; Q.pop(); if(Z > 0) { R.push(i); } } else { return false; } } return true; } int main() { int N = get_int(); int64_t Z = (int64_t)get_int(); priority_queue<pair<int, int> > Qp; priority_queue<pair<int, int> > Qm; for(int i = 1; i <= N; i++) { monsters[i].d = get_int(); monsters[i].a = get_int(); monsters[i].r = monsters[i].a - monsters[i].d; if(monsters[i].r > 0) { Qp.push(pair<int, int>(-monsters[i].d, i)); } else { Qm.push(pair<int, int>(monsters[i].d, i)); } } queue<int> R; bool ret = fight(Qp, Z, R); if(ret) { fight(Qm, Z, R); } if(Z > 0 && R.size() == N) { printf("TAK\n"); while(!R.empty()) { printf("%d ", R.front()); R.pop(); } printf("\n"); } else { printf("NIE\n"); } } |
English