#include<bits/stdc++.h> using namespace std; constexpr int N =5e5+7; int minn[N], maxx[N]; int n,k,x; vector<int> przy; int main() { ios_base::sync_with_stdio(0); cin.tie(0); cout.tie(0); cin >> n >> k; for(int i=0; i<n; i++) { cin >> x; przy.push_back(x); } if(k==2) { minn[0]=przy[0]; for(int i=1; i<n; i++) { minn[i]=min(przy[i],minn[i-1]); } maxx[n-1]=przy[n-1]; for(int i=n-1; i>=0; i--) { maxx[i]=max(przy[i],maxx[i+1]); } for(int i=0; i<n-1; i++) { //cout << minn[i] << " " << maxx[i] << "\n"; if(minn[i]>=maxx[i+1]) { cout << "TAK\n"; cout << i+1; return 0; } } cout << "NIE"; return 0; } if(k==3) { minn[0]=przy[0]; minn[1]=1; for(int i=1; i<n; i++) { if(minn[0]>=przy[i]) { minn[0]=przy[i]; minn[1]=i+1; } } maxx[0]=przy[n-1]; maxx[1]=n; for(int i=n-1; i>=0; i--) { if(maxx[0]<=przy[i]) { maxx[0]=przy[i]; maxx[1]=i+1; } } // cout << minn[0] << " " << minn[1] << "\n"; // cout << maxx[0] << " " << maxx[1] << "\n"; if(maxx[1]!=n) { cout << "TAK\n"; if(maxx[1]==1) maxx[1]++; cout << maxx[1]-1 << " " << maxx[1]; return 0; } if(minn[1]!=1) { cout << "TAK\n"; if(minn[1]==n) minn[1]--; cout << minn[1]-1 << " " << minn[1]; return 0; } cout << "NIE"; return 0; } { //k>=4 x=-1; for(int i=0; i<n-1; i++) { if(przy[i]>=przy[i+1]) { x=i+1; break; } } if(x==1) x++; if(x==n-1) x--; if(x==-1) { cout <<"NIE"; } else { cout << "TAK\n"; if(k-3>=x) { for(int i=1; i<k; i++) cout << i << " "; } else { for(int i=1; i<k-3; i++) cout << i << " "; cout << x-1 << " " << x << " " << x + 1; } } } 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 129 130 131 132 133 134 | #include<bits/stdc++.h> using namespace std; constexpr int N =5e5+7; int minn[N], maxx[N]; int n,k,x; vector<int> przy; int main() { ios_base::sync_with_stdio(0); cin.tie(0); cout.tie(0); cin >> n >> k; for(int i=0; i<n; i++) { cin >> x; przy.push_back(x); } if(k==2) { minn[0]=przy[0]; for(int i=1; i<n; i++) { minn[i]=min(przy[i],minn[i-1]); } maxx[n-1]=przy[n-1]; for(int i=n-1; i>=0; i--) { maxx[i]=max(przy[i],maxx[i+1]); } for(int i=0; i<n-1; i++) { //cout << minn[i] << " " << maxx[i] << "\n"; if(minn[i]>=maxx[i+1]) { cout << "TAK\n"; cout << i+1; return 0; } } cout << "NIE"; return 0; } if(k==3) { minn[0]=przy[0]; minn[1]=1; for(int i=1; i<n; i++) { if(minn[0]>=przy[i]) { minn[0]=przy[i]; minn[1]=i+1; } } maxx[0]=przy[n-1]; maxx[1]=n; for(int i=n-1; i>=0; i--) { if(maxx[0]<=przy[i]) { maxx[0]=przy[i]; maxx[1]=i+1; } } // cout << minn[0] << " " << minn[1] << "\n"; // cout << maxx[0] << " " << maxx[1] << "\n"; if(maxx[1]!=n) { cout << "TAK\n"; if(maxx[1]==1) maxx[1]++; cout << maxx[1]-1 << " " << maxx[1]; return 0; } if(minn[1]!=1) { cout << "TAK\n"; if(minn[1]==n) minn[1]--; cout << minn[1]-1 << " " << minn[1]; return 0; } cout << "NIE"; return 0; } { //k>=4 x=-1; for(int i=0; i<n-1; i++) { if(przy[i]>=przy[i+1]) { x=i+1; break; } } if(x==1) x++; if(x==n-1) x--; if(x==-1) { cout <<"NIE"; } else { cout << "TAK\n"; if(k-3>=x) { for(int i=1; i<k; i++) cout << i << " "; } else { for(int i=1; i<k-3; i++) cout << i << " "; cout << x-1 << " " << x << " " << x + 1; } } } return 0; } |