#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; } } |