Zadanie szkopuł pudełka

Question

Hej, mam problem ze skróceniem działania programu w zadaniu ze szkopuła. Link do zadania:https://szkopul.edu.pl/problemset/problem/pudelka/site/?key=statement

Z góry dziękuje za pomoc.

#include <iostream>
#include <algorithm>
#include <vector>
using namespace std;
int main (){
    ios_base::sync_with_stdio(0);
    cin.tie(0);
    vector<long long> wysokosci(10000001);
    vector<long long>::iterator it;
    long long n, i, p; 
    int m;
    cin >> n;
    for (i=1; i<=n; ++i){
        cin >> wysokosci[i];
    }
    sort (wysokosci.begin()+1, wysokosci.begin()+n+1);
    cin >> m;
    for (i=0; i<m; ++i){
        cin >> p;
        it=lower_bound(wysokosci.begin()+1, wysokosci.begin()+n+1, p);
        if (*it==p){
            cout << it-wysokosci.begin() << '\n';
            }
        else
            cout << "Kup pudelko!" << '\n';
    }
    return 0;
}

 

Answer 1

Przypadek gdy nie znaleziono wartości w podanym zakresie - lower_bound zwróci ostatni iterator:

if (it != wysokosci.begin() + n + 1 and *it == p) {

Kod inaczej:

#include <iostream>
#include <vector>
#include <algorithm>

int main() {
    std::ios_base::sync_with_stdio(false);
    std::cin.tie(nullptr);

    int count;
    std::cin >> count;
    std::vector<int> boxes(count);
    for (auto& box : boxes) {
        std::cin >> box;
    }

    int queries;
    std::cin >> queries;
    while (queries--) {
        int height;
        std::cin >> height;
        auto it = std::lower_bound(boxes.begin(), boxes.end(), height);
        if (it != boxes.end() and *it == height) {
            std::cout << it - boxes.begin() + 1 << "\n";
        } else {
            std::cout << "Kup pudelko!\n";
        }
    }
}