Сдвиг элементов одномерного массива по условию

@Harmonster · Регистрация: 20.05.2012

Студворк — интернет-сервис помощи студентам

Есть функция, она считает количество нулевых элементов (counter) и удаляет его (нулевой элемент). Работает вроде бы правильно, но сдвиг работает неправильно. Подскажите, как исправить?

C++

int *ZeroElementsCounter(int* _array, int n)
{
    int counter = 0, i, k;
    for (i = 0; i < n; i++)
    {
        if (_array[i] == 0)
        {
                _array[i] = _array[i + 1];
                n--;
                counter++;
        }
    }
    return _array;
}

valeriikozlov · 11.01.2014, 22:11

вот так попробуйте:

C++

int *ZeroElementsCounter(int* _array, int n)
{
    int counter = 0, i, k;
    for (i = 0; i < n; i++)
    {
        if (_array[i] == 0)
        {
            counter++;
            for(k=i; k<n-1; k++)
                _array[k] = _array[k + 1];
            n--;
            i--;                
        }
    }
    return _array;
}

@Alexdemath · 12.01.2014, 03:51

Лучше, наверное, без вложенного цикла

C++

int k = 0;
 
for (int i = 0; i < n; i++) 
 { if (_array[i] != 0) _array[k++] = _array[i]; }
 
int counter = n - k;  // Количество нулевых элементов

@outoftime · 12.01.2014, 05:08

C++

#include <iostream>
#include <algorithm>
#include <iterator>
#include <type_traits>
 
template<typename _Type, std::size_t _N>
std::size_t zero_elements_counter(_Type (&array)[_N])
{
    static_assert(std::is_arithmetic<_Type>::value,
                  "Array type must be arithmetic.");
 
    _Type *first = array,
        *last = array + _N,
        *dest = first;
 
    while (first != last)
    {
        if (*first) std::iter_swap(dest++, first++);
        else ++first;
    }
 
    return last - dest;
}
 
int main(int argc, char* argv[])
{
    int a[] = {1,2,3,4,0,1,2,3,4,0};
 
    std::cout << zero_elements_counter(a) << std::endl;
    for (auto e : a)
        std::cout << e << " ";
}

@Harmonster · 12.01.2014, 09:32 **[ТС]**

При вводе n = 5;
Ввожу: 5 0 3 0 9
Вывод: 5 3 9 9 9

Как замененные элементы вообще убрать?

@outoftime · 12.01.2014, 09:41

Harmonster,

Сообщение от outoftime

std::iter_swap(dest++, first++)

вы явно взяли не мой код.

Сообщение от Harmonster

Как замененные элементы вообще убрать?

перевыделить память под массив и вернуть ссылку на новый или использовать std::vector::resize на соответствующем объекте.

@Harmonster · 12.01.2014, 10:11 **[ТС]**

Сообщение от outoftime

Harmonster,
вы явно взяли не мой код.

перевыделить память под массив и вернуть ссылку на новый или использовать std::vector::resize на соответствующем объекте.

Ну почему же? С вашим кодом вывод консоли (см. приложение). В начале массив из 10 элементов, и вывело массив из 10 элементов.

@outoftime · 12.01.2014, 11:25

Сообщение от Harmonster

При вводе n = 5;
Ввожу: 5 0 3 0 9
Вывод: 5 3 9 9 9

У меня нули отбрасываются в конец массива. Возможность обрезать контейнер предоставляю вам. На вопросы отвечу.

Добавлено через 1 минуту
Harmonster, P.S. для форматировая вывода из консоли используйте BB-код: [BASH] также как вы форматируете сишный код.

Добавлено через 1 минуту
К примеру:

Bash

D:\Projects\c++\help\help>g++ -Wall main.cpp -std=c++11
 
D:\Projects\c++\help\help>a.exe
2
1 2 3 4 1 2 3 4 0 0

Добавлено через 15 минут
Не подглядывать

Кликните здесь для просмотра всего текста

C++

#include <iostream>
#include <algorithm>
#include <iterator>
#include <type_traits>
 
template<typename _Type>
size_t zero_elements_counter(std::vector<_Type> &v)
{
    auto dest = v.begin();
    for (auto it = v.begin(); it != v.end(); ++it)
    {
        if (*it) *dest++ = *it;
    }
 
    size_t res = v.end() - dest;
    v.resize(std::distance(v.begin(), dest));
    return res;
}
 
int main(int argc, char* argv[])
{
    std::vector<int> a = {1,2,3,4,0,1,2,3,4,0};
 
    std::cout << zero_elements_counter(a) << std::endl;
    for (auto e : a)
        std::cout << e << " ";
}

Bash

D:\Projects\c++\help\help>g++ -Wall main.cpp -std=c++11
 
D:\Projects\c++\help\help>a.exe
2
1 2 3 4 1 2 3 4

@Harmonster 9 / 9 / 7 Регистрация: 20.05.2012 Сообщений: 101
	12.01.2014, 09:32 [ТС]	5
	При вводе n = 5; Ввожу: 5 0 3 0 9 Вывод: 5 3 9 9 9 Как замененные элементы вообще убрать? 0

@outoftime ║XLR8║ 1212 / 909 / 270 Регистрация: 25.07.2009 Сообщений: 4,361 Записей в блоге: 5
	12.01.2014, 09:41	6
	Harmonster, Сообщение от outoftime std::iter_swap(dest++, first++) вы явно взяли не мой код. Сообщение от Harmonster Как замененные элементы вообще убрать? перевыделить память под массив и вернуть ссылку на новый или использовать std::vector::resize на соответствующем объекте. 0

@Harmonster 9 / 9 / 7 Регистрация: 20.05.2012 Сообщений: 101
	12.01.2014, 10:11 [ТС]	7
	Сообщение от outoftime Harmonster, вы явно взяли не мой код. перевыделить память под массив и вернуть ссылку на новый или использовать std::vector::resize на соответствующем объекте. Ну почему же? С вашим кодом вывод консоли (см. приложение). В начале массив из 10 элементов, и вывело массив из 10 элементов. Миниатюры 0

Опции темы