#include<vector>
#include <iostream>
#include<algorithm>
#include<queue>
#include<bitset>
using namespace std;
typedef long long ll;
#define f first
#define s second
#define pb push_back
int n,k,ile;
bitset<500011> poz;
int A[500001],mini[500001],maxi[500001];
int main()
{ ios_base::sync_with_stdio(); cin.tie();
cin>>n>>k;
ile=k-1;
for(int i=0;i<n;i++)
cin>>A[i];
for(int i=1;i<n;i++)
{if(A[i]<=A[i-1])break;
if(i==n-1){cout<<"NIE"; return 0;}
}
if(k>=4)
{ cout<<"TAK\n";
for(int i=1;i<n;i++)
if(A[i]<=A[i-1]){poz[i-1]=1; poz[i]=1; poz[i+1]=1; ile-=3; if(i==n-1)ile++; if(i==1)ile++; break;}
}
if(k==3)
{
for(int i=1;i<n;i++)if(A[i]<=A[0]){cout<<"TAK\n"; poz[i].flip(); poz[i+1].flip(); ile-=2; if(i==n-1)ile++; break;}
for(int i=n-2;i>=1;i++)if(A[i]>=A[n-1]){cout<<"TAK\n"; poz[i].flip(); poz[i+1].flip(); ile-=2; break;}
}
if(k==2)
{
mini[0]=A[0];
for(int i=1;i<n;i++)
mini[i]=min(mini[i-1],A[i]);
maxi[n-1]=A[n-1];
for(int i=n-2;i>=0;i--)maxi[i]=max(maxi[i+1],A[i]);
for(int i=0;i<n-1;i++)if(mini[i]>=maxi[i+1]){poz[i+1].flip(); ile--; break;}
}
if(ile==k-1){cout<<"NIE"; return 0;}
for(int i=1;i<n;i++){if(ile==0)break; if(poz[i]==0){poz[i].flip(); ile--;}}
for(int i=0;i<n;i++)if(poz[i])cout<<i<<' ';
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 | #include<vector> #include <iostream> #include<algorithm> #include<queue> #include<bitset> using namespace std; typedef long long ll; #define f first #define s second #define pb push_back int n,k,ile; bitset<500011> poz; int A[500001],mini[500001],maxi[500001]; int main() { ios_base::sync_with_stdio(); cin.tie(); cin>>n>>k; ile=k-1; for(int i=0;i<n;i++) cin>>A[i]; for(int i=1;i<n;i++) {if(A[i]<=A[i-1])break; if(i==n-1){cout<<"NIE"; return 0;} } if(k>=4) { cout<<"TAK\n"; for(int i=1;i<n;i++) if(A[i]<=A[i-1]){poz[i-1]=1; poz[i]=1; poz[i+1]=1; ile-=3; if(i==n-1)ile++; if(i==1)ile++; break;} } if(k==3) { for(int i=1;i<n;i++)if(A[i]<=A[0]){cout<<"TAK\n"; poz[i].flip(); poz[i+1].flip(); ile-=2; if(i==n-1)ile++; break;} for(int i=n-2;i>=1;i++)if(A[i]>=A[n-1]){cout<<"TAK\n"; poz[i].flip(); poz[i+1].flip(); ile-=2; break;} } if(k==2) { mini[0]=A[0]; for(int i=1;i<n;i++) mini[i]=min(mini[i-1],A[i]); maxi[n-1]=A[n-1]; for(int i=n-2;i>=0;i--)maxi[i]=max(maxi[i+1],A[i]); for(int i=0;i<n-1;i++)if(mini[i]>=maxi[i+1]){poz[i+1].flip(); ile--; break;} } if(ile==k-1){cout<<"NIE"; return 0;} for(int i=1;i<n;i++){if(ile==0)break; if(poz[i]==0){poz[i].flip(); ile--;}} for(int i=0;i<n;i++)if(poz[i])cout<<i<<' '; return 0; } |
English