#include<iostream>
using namespace std;
int n,k;
int t[600000];
int minn[600000];
int maxx[600000];
bool biore[600000];
void licz(int x)
{
minn[1]=t[1];
for(int i=2;i<=x;i++)
minn[i]=min(minn[i-1],t[i]);
maxx[x+1]=0;
for(int i=x;i>=1;i--)
maxx[i]=max(maxx[i+1],t[i]);
}
void rozw2()
{
licz(n);
for(int i=1;i<n;i++)
{
if(minn[i]>=maxx[i+1])
{
cout<<"TAK\n"<<i;
return;
}
}
cout<<"NIE";
}
void rozw3()
{
licz(n);
int max=maxx[1];
int gdzie=0;
for(int i=2;i<n;i++)
{
if(t[i]==max)
{
gdzie=i;
break;
}
}
if(gdzie!=0)
{
cout<<"TAK\n"<<gdzie-1<<" "<<gdzie;
return;
}
if(t[1]==max)
{
cout<<"TAK\n1 2\n";
return;
}
else{
licz(n-1);
for(int i=1;i<n-1;i++)
{
if(minn[i]>=maxx[i+1])
{
cout<<"TAK\n"<<i<<" "<<n-1;
return;
}
}
}
cout<<"NIE";
}
void rozw4()
{
bool git=0;
int suma=0;
for(int i=1;i<n;i++)
{
if(t[i]>=t[i+1])
{
if(i>1)
biore[i-1]=1;
biore[i]=1;
if(i<n-1)
biore[i+1]=1;
git=1;
suma=biore[i-1]+biore[i]+biore[i+1];
break;
}
}
if(git==0)
{
cout<<"NIE";
return;
}
cout<<"TAK\n";
for(int i=1;i<n;i++)
{
if(biore[i])
cout<<i<<" ";
else if(suma<k-1)
{
suma++;
cout<<i<<" ";
}
}
}
int main()
{
cin.tie(0);
cout.tie(0);
ios_base::sync_with_stdio(0);
cin>>n>>k;
for(int i=1;i<=n;i++)
cin>>t[i];
if(k==2)
rozw2();
else if(k==3)
rozw3();
else rozw4();
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 | #include<iostream> using namespace std; int n,k; int t[600000]; int minn[600000]; int maxx[600000]; bool biore[600000]; void licz(int x) { minn[1]=t[1]; for(int i=2;i<=x;i++) minn[i]=min(minn[i-1],t[i]); maxx[x+1]=0; for(int i=x;i>=1;i--) maxx[i]=max(maxx[i+1],t[i]); } void rozw2() { licz(n); for(int i=1;i<n;i++) { if(minn[i]>=maxx[i+1]) { cout<<"TAK\n"<<i; return; } } cout<<"NIE"; } void rozw3() { licz(n); int max=maxx[1]; int gdzie=0; for(int i=2;i<n;i++) { if(t[i]==max) { gdzie=i; break; } } if(gdzie!=0) { cout<<"TAK\n"<<gdzie-1<<" "<<gdzie; return; } if(t[1]==max) { cout<<"TAK\n1 2\n"; return; } else{ licz(n-1); for(int i=1;i<n-1;i++) { if(minn[i]>=maxx[i+1]) { cout<<"TAK\n"<<i<<" "<<n-1; return; } } } cout<<"NIE"; } void rozw4() { bool git=0; int suma=0; for(int i=1;i<n;i++) { if(t[i]>=t[i+1]) { if(i>1) biore[i-1]=1; biore[i]=1; if(i<n-1) biore[i+1]=1; git=1; suma=biore[i-1]+biore[i]+biore[i+1]; break; } } if(git==0) { cout<<"NIE"; return; } cout<<"TAK\n"; for(int i=1;i<n;i++) { if(biore[i]) cout<<i<<" "; else if(suma<k-1) { suma++; cout<<i<<" "; } } } int main() { cin.tie(0); cout.tie(0); ios_base::sync_with_stdio(0); cin>>n>>k; for(int i=1;i<=n;i++) cin>>t[i]; if(k==2) rozw2(); else if(k==3) rozw3(); else rozw4(); return 0; } |
English