//~ while (clock()<=69*CLOCKS_PER_SEC) //~ #pragma comment(linker, "/stack:200000000") #pragma GCC optimize("O3") //~ #pragma GCC optimize("Ofast") //~ #pragma GCC target("sse,sse2,sse3,ssse3,sse4,popcnt,abm,mmx,avx,tune=native") //~ #pragma GCC optimize("unroll-loops") #include <bits/stdc++.h> using namespace std; using ll=long long; using ld=long double; using pii=pair<int, int>; using pll=pair<ll, ll>; using vi=vector<int>; using vll=vector<ll>; template<class TH> void _dbg(const char *sdbg, TH h){ cerr<<sdbg<<" = "<<h<<endl; } #ifdef LOCAL #define _upgrade ios_base::sync_with_stdio(0); #define debug(...) _dbg(#__VA_ARGS__, __VA_ARGS__); #else #define _upgrade ios_base::sync_with_stdio(0), cin.tie(0), cout.tie(0); #define debug(...) 98; #define cerr if(0) cout #endif #define pb push_back #define mp make_pair #define all(x) x.begin(),x.end() #define sz(x) ((int)(x).size()) #define fi first #define se second #define erase_duplicates(x) sort(all(x)), (x).resize(distance((x).begin(), unique(all(x)))); #define watch(x) cout << (#x) << " is " << (x) << endl; #define pow2(x) ((x)*(x)) #define mod(x, m) ((((x) % (m)) + (m)) % (m)) #define max3(a, b, c) max(a, max(b, c)) #define min3(a, b, c) min(a, min(b, c)) #define REP(i, a, b) for(int i = a; i <= b; ++i) #define REPV(i, b, a) for(int i = b; i >= a; --i) const double m_pi = acos(-1.0); const int int_max = 0x3f3f3f3f; const int M7 = 1000*1000*1000+7; const int M9 = 1000*1000*1000+9; int tab[500'009]; int minleft[500'009]; int maxright[500'009]; int main() { _upgrade mt19937 rng(chrono::steady_clock::now().time_since_epoch().count()); cout<<fixed<<setprecision(15); int n, k; cin>>n>>k; REP(i,0,n-1){ cin>>tab[i]; } vector<int> result; if(k == 2) { minleft[0] = tab[0]; REP(i,1,n-1){ minleft[i] = min(minleft[i-1], tab[i]); } maxright[n-1] = tab[n-1]; REPV(i,n-2,0){ maxright[i] = max(maxright[i+1], tab[i]); } REP(i,0,n-2){ if(minleft[i] >= maxright[i+1]) { if(!sz(result)){ result.pb(i+1); } } } } else if(k == 3) { REP(i,1,n-2){ int minim = tab[0]; int maxim = tab[n-1]; if(tab[i] <= minim || tab[i] >= maxim) { if(!sz(result)) { result.pb(i); result.pb(i+1); } } } } else if(k >= 4) { vector<int> proof; REP(i, 0, n-2) { if(tab[i] >= tab[i+1]) { if(i > 0) { proof.pb(i); } proof.pb(i+1); if(i < n-2) { proof.pb(i+2); } } } if(sz(proof)) { proof.pb(600'000); int proofToAdd = sz(proof); int proofIdx = 0; REP(i, 0, n-2) { if(i + 1 == proof[proofIdx]) { result.pb(i+1); --k; ++proofIdx; --proofToAdd; } else if(k-1-proofToAdd > 0) { result.pb(i+1); --k; } } } } if(sz(result)){ cout << "TAK\n"; REP(i,0,sz(result)-1){ cout<<result[i] << " "; } } else { cout << "NIE\n"; } return 0; } /* __builtin_popcount count total set bits on a given integer (use __builtin_popcountll for long long). */
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 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 | //~ while (clock()<=69*CLOCKS_PER_SEC) //~ #pragma comment(linker, "/stack:200000000") #pragma GCC optimize("O3") //~ #pragma GCC optimize("Ofast") //~ #pragma GCC target("sse,sse2,sse3,ssse3,sse4,popcnt,abm,mmx,avx,tune=native") //~ #pragma GCC optimize("unroll-loops") #include <bits/stdc++.h> using namespace std; using ll=long long; using ld=long double; using pii=pair<int, int>; using pll=pair<ll, ll>; using vi=vector<int>; using vll=vector<ll>; template<class TH> void _dbg(const char *sdbg, TH h){ cerr<<sdbg<<" = "<<h<<endl; } #ifdef LOCAL #define _upgrade ios_base::sync_with_stdio(0); #define debug(...) _dbg(#__VA_ARGS__, __VA_ARGS__); #else #define _upgrade ios_base::sync_with_stdio(0), cin.tie(0), cout.tie(0); #define debug(...) 98; #define cerr if(0) cout #endif #define pb push_back #define mp make_pair #define all(x) x.begin(),x.end() #define sz(x) ((int)(x).size()) #define fi first #define se second #define erase_duplicates(x) sort(all(x)), (x).resize(distance((x).begin(), unique(all(x)))); #define watch(x) cout << (#x) << " is " << (x) << endl; #define pow2(x) ((x)*(x)) #define mod(x, m) ((((x) % (m)) + (m)) % (m)) #define max3(a, b, c) max(a, max(b, c)) #define min3(a, b, c) min(a, min(b, c)) #define REP(i, a, b) for(int i = a; i <= b; ++i) #define REPV(i, b, a) for(int i = b; i >= a; --i) const double m_pi = acos(-1.0); const int int_max = 0x3f3f3f3f; const int M7 = 1000*1000*1000+7; const int M9 = 1000*1000*1000+9; int tab[500'009]; int minleft[500'009]; int maxright[500'009]; int main() { _upgrade mt19937 rng(chrono::steady_clock::now().time_since_epoch().count()); cout<<fixed<<setprecision(15); int n, k; cin>>n>>k; REP(i,0,n-1){ cin>>tab[i]; } vector<int> result; if(k == 2) { minleft[0] = tab[0]; REP(i,1,n-1){ minleft[i] = min(minleft[i-1], tab[i]); } maxright[n-1] = tab[n-1]; REPV(i,n-2,0){ maxright[i] = max(maxright[i+1], tab[i]); } REP(i,0,n-2){ if(minleft[i] >= maxright[i+1]) { if(!sz(result)){ result.pb(i+1); } } } } else if(k == 3) { REP(i,1,n-2){ int minim = tab[0]; int maxim = tab[n-1]; if(tab[i] <= minim || tab[i] >= maxim) { if(!sz(result)) { result.pb(i); result.pb(i+1); } } } } else if(k >= 4) { vector<int> proof; REP(i, 0, n-2) { if(tab[i] >= tab[i+1]) { if(i > 0) { proof.pb(i); } proof.pb(i+1); if(i < n-2) { proof.pb(i+2); } } } if(sz(proof)) { proof.pb(600'000); int proofToAdd = sz(proof); int proofIdx = 0; REP(i, 0, n-2) { if(i + 1 == proof[proofIdx]) { result.pb(i+1); --k; ++proofIdx; --proofToAdd; } else if(k-1-proofToAdd > 0) { result.pb(i+1); --k; } } } } if(sz(result)){ cout << "TAK\n"; REP(i,0,sz(result)-1){ cout<<result[i] << " "; } } else { cout << "NIE\n"; } return 0; } /* __builtin_popcount count total set bits on a given integer (use __builtin_popcountll for long long). */ |