#include<iostream>
#include<cmath>
#include<vector>
#include<cstdlib>
using namespace std;
int a,b,najw=0,il_najw=0,najm=500001,il_najm=0;
vector<int> indexy_najw, indexy_najm;
int main()
{
scanf("%d",&a);
scanf("%d",&b);
int A[a];
int z;
if (b!=2)
{
for (int i=0; i<a; i++)
{
scanf("%d",&A[i]);
if (A[i]==najw)
{
il_najw++;
indexy_najw.push_back(i);
}
else if (A[i] > najw)
{
najw = A[i];
il_najw = 1;
indexy_najw.clear();
indexy_najw.push_back(i);
}
if (A[i]==najm)
{
il_najm++;
indexy_najm.push_back(i);
}
else if (A[i] < najm)
{
najm = A[i];
il_najm = 1;
indexy_najm.clear();
indexy_najm.push_back(i);
}
}
for (int i = 0;i<il_najw;i++)
{
if (indexy_najw[i]!=a-1 )
{
printf("%s\n", "TAK");
for (z=1;z < b-2 && z < indexy_najw[i]; z++)
printf("%d ",z);
if (indexy_najw[i]!=0)
cout << indexy_najw[i] << " ";
cout << indexy_najw[i]+1 << " ";
for (int j=indexy_najw[i]+2; j < b;j++)
printf("%d ",j);
return 0;
}
}
for (int i = 0;i<il_najm;i++)
{
if (indexy_najm[i]!=0)
{
printf("%s\n", "TAK");
for (z=1;z < b-2 && z < indexy_najm[i]; z++)
printf("%d ",z);
cout << indexy_najm[i] << " ";
cout << indexy_najm[i]+1 << " ";
for (int j=indexy_najm[i]+2; j < b;j++)
printf("%d ",j);
return 0;
}
}
if (b < 3)
{
printf("%s", "NIE");
return 0;
}
int aknajw=najw,aknajm=najm;
indexy_najm.clear();
indexy_najw.clear();
for (int powt_bez_kon = 1;;powt_bez_kon++)
{
int aknajw=najw,aknajm=najm;
najw=0,il_najw=0,najm=500001,il_najm=0;
for (int i = 0; i < a ;i++)
{
if (A[i]==najw)
{
il_najw++;
indexy_najw.push_back(i);
}
else if (A[i] < aknajw && A[i] > najw)
{
najw = A[i];
il_najw = 1;
indexy_najw.clear();
indexy_najw.push_back(i);
}
if (A[i]==najm)
{
il_najm++;
indexy_najm.push_back(i);
}
else if (A[i] > aknajm && A[i] < najm)
{
najm = A[i];
il_najm = 1;
indexy_najm.clear();
indexy_najm.push_back(i);
}
}
if (najm >= najw)
{
printf("%s", "NIE");
return 0;
}
for (int i = 0;b > 3 && i<il_najw;i++)
{
if (indexy_najw[i] < a-1-powt_bez_kon)
{
printf("%s\n", "TAK");
for (z=1;z < b-3 && z < indexy_najw[i]; z++)
printf("%d ",z);
if (indexy_najw[i]!=0)
cout << indexy_najw[i] << " ";
cout << indexy_najw[i]+1 << " ";
for (int j=0; j < b-z-2;j++)
printf("%d ",(j+indexy_najw[i]+2));
return 0;
}
}
for (int i = 0;b > 2 && i<il_najm;i++)
{
if (indexy_najm[i] > 0+powt_bez_kon)
{
printf("%s\n", "TAK");
for (z=1;z < b-2 && z < indexy_najm[i]; z++)
printf("%d ",z);
cout << indexy_najm[i] << " ";
cout << indexy_najm[i]+1 << " ";
for (int j=indexy_najm[i]+2; j < b;j++)
printf("%d ",j);
return 0;
}
}
}
}
else
{
for (int i = 0; i < a; i++)
scanf("%d", &A[i]);
for (int i = 1;i < a;i++)
{
int najmnej=500001,najwecej=0;
for (int j = 0; j < i; j++)
if (A[j] < najmnej)
najmnej = A[j];
for (int j = i; j < a; j++)
if (A[j] > najwecej)
najwecej = A[j];
if (najmnej >= najwecej)
{
printf("%s\n", "TAK");
printf("%d", i);
return 0;
}
}
printf("%s", "NIE");
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 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 | #include<iostream> #include<cmath> #include<vector> #include<cstdlib> using namespace std; int a,b,najw=0,il_najw=0,najm=500001,il_najm=0; vector<int> indexy_najw, indexy_najm; int main() { scanf("%d",&a); scanf("%d",&b); int A[a]; int z; if (b!=2) { for (int i=0; i<a; i++) { scanf("%d",&A[i]); if (A[i]==najw) { il_najw++; indexy_najw.push_back(i); } else if (A[i] > najw) { najw = A[i]; il_najw = 1; indexy_najw.clear(); indexy_najw.push_back(i); } if (A[i]==najm) { il_najm++; indexy_najm.push_back(i); } else if (A[i] < najm) { najm = A[i]; il_najm = 1; indexy_najm.clear(); indexy_najm.push_back(i); } } for (int i = 0;i<il_najw;i++) { if (indexy_najw[i]!=a-1 ) { printf("%s\n", "TAK"); for (z=1;z < b-2 && z < indexy_najw[i]; z++) printf("%d ",z); if (indexy_najw[i]!=0) cout << indexy_najw[i] << " "; cout << indexy_najw[i]+1 << " "; for (int j=indexy_najw[i]+2; j < b;j++) printf("%d ",j); return 0; } } for (int i = 0;i<il_najm;i++) { if (indexy_najm[i]!=0) { printf("%s\n", "TAK"); for (z=1;z < b-2 && z < indexy_najm[i]; z++) printf("%d ",z); cout << indexy_najm[i] << " "; cout << indexy_najm[i]+1 << " "; for (int j=indexy_najm[i]+2; j < b;j++) printf("%d ",j); return 0; } } if (b < 3) { printf("%s", "NIE"); return 0; } int aknajw=najw,aknajm=najm; indexy_najm.clear(); indexy_najw.clear(); for (int powt_bez_kon = 1;;powt_bez_kon++) { int aknajw=najw,aknajm=najm; najw=0,il_najw=0,najm=500001,il_najm=0; for (int i = 0; i < a ;i++) { if (A[i]==najw) { il_najw++; indexy_najw.push_back(i); } else if (A[i] < aknajw && A[i] > najw) { najw = A[i]; il_najw = 1; indexy_najw.clear(); indexy_najw.push_back(i); } if (A[i]==najm) { il_najm++; indexy_najm.push_back(i); } else if (A[i] > aknajm && A[i] < najm) { najm = A[i]; il_najm = 1; indexy_najm.clear(); indexy_najm.push_back(i); } } if (najm >= najw) { printf("%s", "NIE"); return 0; } for (int i = 0;b > 3 && i<il_najw;i++) { if (indexy_najw[i] < a-1-powt_bez_kon) { printf("%s\n", "TAK"); for (z=1;z < b-3 && z < indexy_najw[i]; z++) printf("%d ",z); if (indexy_najw[i]!=0) cout << indexy_najw[i] << " "; cout << indexy_najw[i]+1 << " "; for (int j=0; j < b-z-2;j++) printf("%d ",(j+indexy_najw[i]+2)); return 0; } } for (int i = 0;b > 2 && i<il_najm;i++) { if (indexy_najm[i] > 0+powt_bez_kon) { printf("%s\n", "TAK"); for (z=1;z < b-2 && z < indexy_najm[i]; z++) printf("%d ",z); cout << indexy_najm[i] << " "; cout << indexy_najm[i]+1 << " "; for (int j=indexy_najm[i]+2; j < b;j++) printf("%d ",j); return 0; } } } } else { for (int i = 0; i < a; i++) scanf("%d", &A[i]); for (int i = 1;i < a;i++) { int najmnej=500001,najwecej=0; for (int j = 0; j < i; j++) if (A[j] < najmnej) najmnej = A[j]; for (int j = i; j < a; j++) if (A[j] > najwecej) najwecej = A[j]; if (najmnej >= najwecej) { printf("%s\n", "TAK"); printf("%d", i); return 0; } } printf("%s", "NIE"); return 0; } } |
English