#include<bits/stdc++.h>
using namespace std;
int n,k,a[500001],najw[500002];
bool ck[500001];
int main()
{
scanf("%d%d",&n,&k);
a[0]=INT_MIN;
int gdzie=0;
for(int i=1;i<=n;++i)
{
scanf("%d",&a[i]);
if(a[i]<=a[i-1])
gdzie=i;
}
if(gdzie==0)
{
printf("NIE");
return 0;
}
if(k==2)
{
najw[n+1]=INT_MIN;
for(int i=n;i>1;--i)
najw[i]=max(najw[i+1],a[i]);
int najm=INT_MAX;
for(int i=1;i<n;++i)
{
najm=min(najm,a[i]);
if(najm>=najw[i+1])
{
printf("TAK\n%d",i);
return 0;
}
}
printf("NIE");
}
else if(k==3)
{
for(int i=n;i>1;--i)
najw[i]=max(najw[i+1],a[i]);
int najm=a[1];
for(int i=2;i<n;++i)
{
if(a[i]>=najw[i+1]||najm>=a[i])
{
printf("TAK\n%d %d",i-1,i);
return 0;
}
najm=min(najm,a[i]);
}
printf("NIE");
}
else
{
if(gdzie==2)
{
ck[1]=1;
ck[2]=1;
k-=2;
for(int i=3;k>1;++i)
{
--k;
ck[i]=1;
}
}
else if(gdzie==n)
{
ck[n-1]=1;
k-=2;
for(int i=n-2;k;--i)
{
--k;
ck[i]=1;
}
}
else
{
k-=2;
int lew=gdzie-2,praw=gdzie+1;
ck[gdzie]=1;
ck[gdzie-1]=1;
while(lew>1&&k>=2)
{
--k;
ck[lew]=1;
--lew;
}
while(praw<n&&k>=2)
{
--k;
ck[praw]=1;
++praw;
}
ck[lew]=1;
}
printf("TAK\n");
for(int i=1;i<=n;++i)
if(ck[i])
printf("%d ",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 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 | #include<bits/stdc++.h> using namespace std; int n,k,a[500001],najw[500002]; bool ck[500001]; int main() { scanf("%d%d",&n,&k); a[0]=INT_MIN; int gdzie=0; for(int i=1;i<=n;++i) { scanf("%d",&a[i]); if(a[i]<=a[i-1]) gdzie=i; } if(gdzie==0) { printf("NIE"); return 0; } if(k==2) { najw[n+1]=INT_MIN; for(int i=n;i>1;--i) najw[i]=max(najw[i+1],a[i]); int najm=INT_MAX; for(int i=1;i<n;++i) { najm=min(najm,a[i]); if(najm>=najw[i+1]) { printf("TAK\n%d",i); return 0; } } printf("NIE"); } else if(k==3) { for(int i=n;i>1;--i) najw[i]=max(najw[i+1],a[i]); int najm=a[1]; for(int i=2;i<n;++i) { if(a[i]>=najw[i+1]||najm>=a[i]) { printf("TAK\n%d %d",i-1,i); return 0; } najm=min(najm,a[i]); } printf("NIE"); } else { if(gdzie==2) { ck[1]=1; ck[2]=1; k-=2; for(int i=3;k>1;++i) { --k; ck[i]=1; } } else if(gdzie==n) { ck[n-1]=1; k-=2; for(int i=n-2;k;--i) { --k; ck[i]=1; } } else { k-=2; int lew=gdzie-2,praw=gdzie+1; ck[gdzie]=1; ck[gdzie-1]=1; while(lew>1&&k>=2) { --k; ck[lew]=1; --lew; } while(praw<n&&k>=2) { --k; ck[praw]=1; ++praw; } ck[lew]=1; } printf("TAK\n"); for(int i=1;i<=n;++i) if(ck[i]) printf("%d ",i); } return 0; } |
English