#include <iostream> #include <vector> #include <algorithm> using punktacja = std::vector<int>; using std::cout; using std::endl; using std::cin; punktacja wczytajDane(int &uczestnicy, int &koszulki) { cin >> uczestnicy >> koszulki; punktacja punkty; int wynik; for (int i = 0; i < uczestnicy; i++) { cin >> wynik; punkty.push_back(wynik); } return punkty; } void wypiszPunkty(int n, int k, punktacja p) { cout << n << " " << k << endl; for (int i = 0; i < n; i++) { cout << p[i] << " "; } cout << endl; } punktacja sumyPrefiksowe(punktacja p, int n) { punktacja sumy; sumy.push_back(p[n - 1]); int pop; for (int i = n - 2; i >= 0; i--) { pop = sumy[i + 1]; sumy.push_back(pop + p[i]); } return sumy; } int znajdzLiczbe(punktacja &p, int n, int k) { if (k == 0) return 0; std::sort(p.begin(), p.end()); /*int ostatniaPrawilnaKoszulka = n - 1 - k; int wartosc = p[ostatniaPrawilnaKoszulka]; wypiszPunkty(n, k, p); int index = ostatniaPrawilnaKoszulka; //- 1; cout << p[ostatniaPrawilnaKoszulka] << endl; while (index >= 0 && p[index] == wartosc) { cout << p[index] << endl; index--; } cout << "n: " << n - 1 << " " << index << endl; return n - index - 1; //- 1; */ int ostatniaKoszulka = n - k;//- 1 - k; int wartosc = p[ostatniaKoszulka]; int index = ostatniaKoszulka; while (index >= 0 && p[index] == wartosc) { index--; } //cout << index << " " << wartosc << " " << ostatniaKoszulka << endl; //index = (p[index] == wartosc ? index : (index + 1)); //cout << index << " " << n << endl; return n - index - 1; } int main() { int n, k; punktacja punkty = wczytajDane(n, k); //wypiszPunkty(n, k, punkty); int ile = znajdzLiczbe(punkty, n, k); cout << ile << endl; 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 | #include <iostream> #include <vector> #include <algorithm> using punktacja = std::vector<int>; using std::cout; using std::endl; using std::cin; punktacja wczytajDane(int &uczestnicy, int &koszulki) { cin >> uczestnicy >> koszulki; punktacja punkty; int wynik; for (int i = 0; i < uczestnicy; i++) { cin >> wynik; punkty.push_back(wynik); } return punkty; } void wypiszPunkty(int n, int k, punktacja p) { cout << n << " " << k << endl; for (int i = 0; i < n; i++) { cout << p[i] << " "; } cout << endl; } punktacja sumyPrefiksowe(punktacja p, int n) { punktacja sumy; sumy.push_back(p[n - 1]); int pop; for (int i = n - 2; i >= 0; i--) { pop = sumy[i + 1]; sumy.push_back(pop + p[i]); } return sumy; } int znajdzLiczbe(punktacja &p, int n, int k) { if (k == 0) return 0; std::sort(p.begin(), p.end()); /*int ostatniaPrawilnaKoszulka = n - 1 - k; int wartosc = p[ostatniaPrawilnaKoszulka]; wypiszPunkty(n, k, p); int index = ostatniaPrawilnaKoszulka; //- 1; cout << p[ostatniaPrawilnaKoszulka] << endl; while (index >= 0 && p[index] == wartosc) { cout << p[index] << endl; index--; } cout << "n: " << n - 1 << " " << index << endl; return n - index - 1; //- 1; */ int ostatniaKoszulka = n - k;//- 1 - k; int wartosc = p[ostatniaKoszulka]; int index = ostatniaKoszulka; while (index >= 0 && p[index] == wartosc) { index--; } //cout << index << " " << wartosc << " " << ostatniaKoszulka << endl; //index = (p[index] == wartosc ? index : (index + 1)); //cout << index << " " << n << endl; return n - index - 1; } int main() { int n, k; punktacja punkty = wczytajDane(n, k); //wypiszPunkty(n, k, punkty); int ile = znajdzLiczbe(punkty, n, k); cout << ile << endl; return 0; } |