/****************************************************************************** Online C++ Compiler. Code, Compile, Run and Debug C++ program online. Write your code in this editor and press "Run" button to compile and execute it. *******************************************************************************/ #include <iostream> #include <set> using namespace std; const int MAXN = 500000; int n, k; int a[MAXN]; void do_case_2() { multiset<int> left, right; left.insert(a[0]); for(int i = 1; i < n; i++) right.insert(a[i]); for(int i = 1; i < n; i++) { if( *(left.begin()) >= *(right.rbegin()) ) { cout << "TAK" << endl << i << endl; return; } left.insert(a[i]); right.erase(right.find(a[i])); } cout << "NIE" << endl; } void do_case_3() { for(int i = 1; i < n-1; i++) { if(a[0] >= a[i]) { cout << "TAK" << endl << i << " " << i+1 << endl; return; } } for(int i = n-2; i>0; i--) { if(a[n-1] <= a[i]) { cout << "TAK" << endl << i << " " << i+1 << endl; return; } } cout << "NIE" << endl; } void do_case_many() { for(int i = 1; i < n; i++) { if(a[i-1] >= a[i]) { set<int> breaks; if(i > 1) breaks.insert(i-1); breaks.insert(i); if(i+1 < n) breaks.insert(i+1); for(int j = 1; breaks.size()<k-1; j++) breaks.insert(j); cout << "TAK" << endl; for(auto it = breaks.begin(); it!= breaks.end(); it++) cout << *it << " "; cout << endl; return; } } cout << "NIE" << endl; } int main() { cin >> n >> k; for(int i=0; i < n; i++) cin >> a[i]; if(k==2) do_case_2(); else if(k==3) do_case_3(); else do_case_many(); 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 | /****************************************************************************** Online C++ Compiler. Code, Compile, Run and Debug C++ program online. Write your code in this editor and press "Run" button to compile and execute it. *******************************************************************************/ #include <iostream> #include <set> using namespace std; const int MAXN = 500000; int n, k; int a[MAXN]; void do_case_2() { multiset<int> left, right; left.insert(a[0]); for(int i = 1; i < n; i++) right.insert(a[i]); for(int i = 1; i < n; i++) { if( *(left.begin()) >= *(right.rbegin()) ) { cout << "TAK" << endl << i << endl; return; } left.insert(a[i]); right.erase(right.find(a[i])); } cout << "NIE" << endl; } void do_case_3() { for(int i = 1; i < n-1; i++) { if(a[0] >= a[i]) { cout << "TAK" << endl << i << " " << i+1 << endl; return; } } for(int i = n-2; i>0; i--) { if(a[n-1] <= a[i]) { cout << "TAK" << endl << i << " " << i+1 << endl; return; } } cout << "NIE" << endl; } void do_case_many() { for(int i = 1; i < n; i++) { if(a[i-1] >= a[i]) { set<int> breaks; if(i > 1) breaks.insert(i-1); breaks.insert(i); if(i+1 < n) breaks.insert(i+1); for(int j = 1; breaks.size()<k-1; j++) breaks.insert(j); cout << "TAK" << endl; for(auto it = breaks.begin(); it!= breaks.end(); it++) cout << *it << " "; cout << endl; return; } } cout << "NIE" << endl; } int main() { cin >> n >> k; for(int i=0; i < n; i++) cin >> a[i]; if(k==2) do_case_2(); else if(k==3) do_case_3(); else do_case_many(); return 0; } |