Как запомнить индексы равных элементов массива?

    int max = B[0];
    for(int i=1;i<N;i++)
    {
        if(B[i]>max)
        {
            max = B[i];
            index_i = i;
        }
    }
 
    return max

int max = A[0];
int count;
for(int i=1;i<N;i++)
{
    if(A[i] > max)
    {
        count = 0;
        max = A[i];
        B[count] = i;
        count++;
    }
    else if(A[i] == max)
    {
        B[count] = i;
        count++;
    }
}

if(A[i] > max) {
    max = A[i];
    while ( count > 0 ) {
        B[ i ] = 0;
        count--;
    }
    B[count] = i;
    count++;
}

int max = A[0];
int count = 0;  //вот здесь надо присвоить,иначе во второй ветке else неизвестно чему равен count
B[0] = 0;           //т.к. первый элемент типа уже max, то запомнить его индекс
for(int i=1;i<N;i++)
{
    if(A[i] > max)
    {
        count = 0;
        max = A[i];
        B[count] = i;
        count++;
    }
    else if(A[i] == max)
    {
        if(count == 0)//если первый элемент массива был максимальный
            count++;
        B[count] = i;
        count++;
    }
}