Помогите в c++ Мой код выдает ошибку на одном из тестов

Question

Помогите в c++ Мой код выдает ошибку на одном из тестов

Пётр Попов Профи (511), закрыт 1 месяц назад

Входные данные
В первой строке вводится кол-во элементов в массиве.

Во второй строке вводится массив.

Выходные данные
В первой строке выведите одно число
k
— количество искомых элементов.

Во второй строке выведите
k
чисел — индексы этих элементов в возрастающем порядке.

#include <iostream>
#include <vector>
#include <algorithm>
using namespace std;
int main() {
int counter = 0;
int a = 0;
cin >> a;
vector<int> b;
int x;
vector<int> answer;
for(int i = 0; i<a; i++ ){
cin>>x;
b.push_back(x);
}
if(b[0] > b[1]){
answer.push_back(0);
counter +=1;
}
for(int i = 1; i<a-1; i++){
if(b[i]>b[i-1] && b[i]>b[i+1]){
answer.push_back(i);
counter += 1;
}

}
if(b[a-1] > b[a-2]){
answer.push_back(a-1);
counter +=1;
}
cout<<counter<<"\n";
sort(begin(answer), end(answer));
for(int i = 0; i<answer.size(); i++){
cout<<answer[i]<<" ";
}
}

Answer 1

Андрей Высший разум (472484) 1 месяц назад

А если длина массива - 1 элемент? Или 0 элементов? В этих случаях твой код будет работать заведомо ошибочно.

Николай ВеселухаВысший разум (374407) 1 месяц назад

А что ищем-то?

Андрей Высший разум (472484) Николай Веселуха, Судя по коду, локальные максимумы.

Пётр ПоповПрофи (511) 1 месяц назад

Спасибо! Я не заметил этот ответ, переписал код и спустя час мучений додумался наконец-то. В этом-то и была причина

Answer 2

На не благодари

 #include <iostream> 
#include <vector> 
#include <algorithm> 
 
int main() { 
    int n; 
    std::cin >> n; 
     
    if (n <= 0) return 0; 
     
    std::vector<int> nums(n); 
    for (int& x : nums) { 
        std::cin >> x; 
    } 
     
    std::vector<int> peaks; 
    peaks.reserve(n); 
     
    if (n > 1 && nums[0] > nums[1]) { 
        peaks.push_back(0); 
    } 
     
    for (int i = 1; i < n - 1; ++i) { 
        if (nums[i] > nums[i-1] && nums[i] > nums[i+1]) { 
            peaks.push_back(i); 
        } 
    } 
     
    if (n > 1 && nums[n-1] > nums[n-2]) { 
        peaks.push_back(n-1); 
    } 
     
    std::cout << peaks.size() << '\n'; 
    for (int peak : peaks) { 
        std::cout << peak << ' '; 
    } 
     
    return 0; 
}

Answer 3

 #include <iostream> 
#include <vector>

using namespace std;

int main() { 
    size_t n; 
    cin >> n; 
    if (!n) exit(0); 
    vector<int> v(n); 
    for (auto& x : v) cin >> x; 
    size_t i = 0; 
    const size_t m = n - 1; 
    while (++i < m) { 
        if (v[i - 1] < v[i] && v[i + 1] < v[i]) { 
            cout << i << ' '; 
        } 
    } 
    cout.put('\n'); 
}