#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
typedef pair<ll,ll> pll;
typedef pair<int,int> pii;
typedef vector<ll> vl;
typedef vector<int> vi;
typedef vector<pll> vll;
typedef vector<pii> vii;
typedef double db;
#define pb push_back
#define mp make_pair
#define all(x) x.begin(),x.end()
#define sz(x) (int)x.size()
#define BOOST ios_base::sync_with_stdio(0);cin.tie(0);cout.tie(0);
#define st first
#define nd second
#define FOR(i,s,k) for(auto i=s; i<k; i++)
#define ROF(i,s,k) for(auto i=k-1; i>=s; i--)
constexpr ll inf = 1e18+1;
constexpr ll MAXN = 1e6+5;
constexpr ll LOGN = 18;
int main()
{
BOOST;
int n,k;
cin>>n>>k;
vi V(n);
FOR(i,0,n) cin>>V[i];
bool rosnacy = true;
int poz = -1;
FOR(i,1,n) if(rosnacy && V[i-1] >= V[i])
{
rosnacy = false;
poz = i-1;
}
if(rosnacy)
{
cout<<"NIE";
return 0;
}
if(k>=4)
{
cout<<"TAK\n";
//poz-1, poz, poz+1 o ile któreś z nich nie jest poza przedziałem
if(poz-1 >= 0 && poz+2 < n)
{
FOR(i,1,poz) if(k>4)
{
cout<<i<<" ";
k--;
}//wypisz tak dużo, żeby wystarczyło na te 3
cout<<poz<<" "<<poz+1<<" "<<poz+2<<" ";
k-=3;
FOR(i,poz+3,n+1) if(k>1)//wypisz do końca ile wlezie
{
cout<<i<<" ";
k--;
}
}
else if(poz-1<0)//jeśli to jest przyklejone do początku
{
FOR(i,1,k) cout<<i<<" ";
}
else if(poz+2 == n)//jeśli przyklejone do końca
{
ROF(i,1,k) cout<<n-i<<" ";
}
return 0;
}
else if(k==2)
{
vi pref(n);//minimum na prefiksie
vi suf(n);//maksimum na sufiksie
pref[0] = V[0];
FOR(i,1,n) pref[i] = min(pref[i-1],V[i]);
suf[n-1] = V[n-1];
ROF(i,0,n-1) suf[i] = max(suf[i+1],V[i]);
FOR(i,0,n-1)
{
if(pref[i]>=suf[i+1])
{
cout<<"TAK\n"<<i+1;
return 0;
}
}
cout<<"NIE";
return 0;
}
else if(k==3)
{
int mini = 0, maxi = 0;
FOR(i,0,n)
{
if(V[i] > V[maxi]) maxi = i;
if(V[i] <= V[mini]) mini = i;
}
if(mini == 0 && maxi == n-1)
{
cout<<"NIE";
}
else
{
cout<<"TAK\n";
if(mini!=0)
{
if(mini==n-1)
cout<<mini-1<<" "<<mini;
else
cout<<mini<<" "<<mini+1;
}
else if(maxi != n-1)
{
if(maxi==0)
cout<<maxi+1<<" "<<maxi+2;
else
cout<<maxi<<" "<<maxi+1;
}
}
}
assert(true);
}
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 | #include <bits/stdc++.h> using namespace std; typedef long long ll; typedef pair<ll,ll> pll; typedef pair<int,int> pii; typedef vector<ll> vl; typedef vector<int> vi; typedef vector<pll> vll; typedef vector<pii> vii; typedef double db; #define pb push_back #define mp make_pair #define all(x) x.begin(),x.end() #define sz(x) (int)x.size() #define BOOST ios_base::sync_with_stdio(0);cin.tie(0);cout.tie(0); #define st first #define nd second #define FOR(i,s,k) for(auto i=s; i<k; i++) #define ROF(i,s,k) for(auto i=k-1; i>=s; i--) constexpr ll inf = 1e18+1; constexpr ll MAXN = 1e6+5; constexpr ll LOGN = 18; int main() { BOOST; int n,k; cin>>n>>k; vi V(n); FOR(i,0,n) cin>>V[i]; bool rosnacy = true; int poz = -1; FOR(i,1,n) if(rosnacy && V[i-1] >= V[i]) { rosnacy = false; poz = i-1; } if(rosnacy) { cout<<"NIE"; return 0; } if(k>=4) { cout<<"TAK\n"; //poz-1, poz, poz+1 o ile któreś z nich nie jest poza przedziałem if(poz-1 >= 0 && poz+2 < n) { FOR(i,1,poz) if(k>4) { cout<<i<<" "; k--; }//wypisz tak dużo, żeby wystarczyło na te 3 cout<<poz<<" "<<poz+1<<" "<<poz+2<<" "; k-=3; FOR(i,poz+3,n+1) if(k>1)//wypisz do końca ile wlezie { cout<<i<<" "; k--; } } else if(poz-1<0)//jeśli to jest przyklejone do początku { FOR(i,1,k) cout<<i<<" "; } else if(poz+2 == n)//jeśli przyklejone do końca { ROF(i,1,k) cout<<n-i<<" "; } return 0; } else if(k==2) { vi pref(n);//minimum na prefiksie vi suf(n);//maksimum na sufiksie pref[0] = V[0]; FOR(i,1,n) pref[i] = min(pref[i-1],V[i]); suf[n-1] = V[n-1]; ROF(i,0,n-1) suf[i] = max(suf[i+1],V[i]); FOR(i,0,n-1) { if(pref[i]>=suf[i+1]) { cout<<"TAK\n"<<i+1; return 0; } } cout<<"NIE"; return 0; } else if(k==3) { int mini = 0, maxi = 0; FOR(i,0,n) { if(V[i] > V[maxi]) maxi = i; if(V[i] <= V[mini]) mini = i; } if(mini == 0 && maxi == n-1) { cout<<"NIE"; } else { cout<<"TAK\n"; if(mini!=0) { if(mini==n-1) cout<<mini-1<<" "<<mini; else cout<<mini<<" "<<mini+1; } else if(maxi != n-1) { if(maxi==0) cout<<maxi+1<<" "<<maxi+2; else cout<<maxi<<" "<<maxi+1; } } } assert(true); } |
English