#include<bits/stdc++.h> using namespace std; constexpr int maxn = 1e5+1; int n,k; int ileminprz; int poprz, akt; vector<int> liczby(maxn); int main() { ios_base::sync_with_stdio(0); cin.tie(0); cout.tie(0); cin>>n>>k; if(k == 2) { vector<int> maxr(maxn); vector<int> minl(maxn); cin>>liczby[1]; minl[1] = liczby[1]; for(int i=2;i<=n;i++) { cin>>liczby[i]; minl[i] = min(minl[i-1], liczby[i]); } maxr[n] = liczby[n]; for(int i = n-1;i > 0;i--) { maxr[i] = max(maxr[i+1], liczby[i+1]); if(minl[i] >= maxr[i]) { cout<<"TAK"; cout<<i; return 0; } } cout<<"NIE"; return 0; } if(k == 3) { vector<int> maxr(maxn); vector<pair<int,int>> maxl(maxn); vector<pair<int,int>> minl(maxn); cin>>liczby[1]; maxl[1] = {liczby[1], 1}; minl[1] = {INT_MAX, 1}; for(int i = 2;i <= n;i++) { cin>>liczby[i]; if(liczby[i] <= liczby[1]) { cout<<"TAK\n"; cout<<i-1<<' '<<i; return 0; } if(minl[i-1].first > liczby[i]) minl[i] = {liczby[i], i}; else minl[i] = minl[i-1]; if(maxl[i-1].first <= liczby[i]) maxl[i] = {liczby[i], i}; else maxl[i] = maxl[i-1]; } for(int i = n-1;i > 1;i--) { maxr[i] = max(maxr[i+1], liczby[i+1]); if(minl[i].first >= maxr[i]) { cout<<"TAK"; cout<<minl[i].second-1<<' '<<minl[i].second; return 0; } if(maxl[i].first >= maxr[i]) { cout<<"TAK"; cout<<maxl[i].second-1<<' '<<maxl[i].second; return 0; } } cout<<"NIE"; } if(k > 3) { cin>>poprz; for(int i=1;i<n;i++) { cin>>akt; if(akt <= poprz) { cout<<"TAK\n"; for(int l=1;k>4;k--,l++) cout<<l<<' '; cout<<i-1<<' '<<i<<' '; k-=2; for(int l = i+1;k>1;k--,l++) cout<<l<<' '; return 0; } poprz = akt; } cout<<"NIE"; } 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 | #include<bits/stdc++.h> using namespace std; constexpr int maxn = 1e5+1; int n,k; int ileminprz; int poprz, akt; vector<int> liczby(maxn); int main() { ios_base::sync_with_stdio(0); cin.tie(0); cout.tie(0); cin>>n>>k; if(k == 2) { vector<int> maxr(maxn); vector<int> minl(maxn); cin>>liczby[1]; minl[1] = liczby[1]; for(int i=2;i<=n;i++) { cin>>liczby[i]; minl[i] = min(minl[i-1], liczby[i]); } maxr[n] = liczby[n]; for(int i = n-1;i > 0;i--) { maxr[i] = max(maxr[i+1], liczby[i+1]); if(minl[i] >= maxr[i]) { cout<<"TAK"; cout<<i; return 0; } } cout<<"NIE"; return 0; } if(k == 3) { vector<int> maxr(maxn); vector<pair<int,int>> maxl(maxn); vector<pair<int,int>> minl(maxn); cin>>liczby[1]; maxl[1] = {liczby[1], 1}; minl[1] = {INT_MAX, 1}; for(int i = 2;i <= n;i++) { cin>>liczby[i]; if(liczby[i] <= liczby[1]) { cout<<"TAK\n"; cout<<i-1<<' '<<i; return 0; } if(minl[i-1].first > liczby[i]) minl[i] = {liczby[i], i}; else minl[i] = minl[i-1]; if(maxl[i-1].first <= liczby[i]) maxl[i] = {liczby[i], i}; else maxl[i] = maxl[i-1]; } for(int i = n-1;i > 1;i--) { maxr[i] = max(maxr[i+1], liczby[i+1]); if(minl[i].first >= maxr[i]) { cout<<"TAK"; cout<<minl[i].second-1<<' '<<minl[i].second; return 0; } if(maxl[i].first >= maxr[i]) { cout<<"TAK"; cout<<maxl[i].second-1<<' '<<maxl[i].second; return 0; } } cout<<"NIE"; } if(k > 3) { cin>>poprz; for(int i=1;i<n;i++) { cin>>akt; if(akt <= poprz) { cout<<"TAK\n"; for(int l=1;k>4;k--,l++) cout<<l<<' '; cout<<i-1<<' '<<i<<' '; k-=2; for(int l = i+1;k>1;k--,l++) cout<<l<<' '; return 0; } poprz = akt; } cout<<"NIE"; } return 0; } |