Заменить нулевые значения элементов массива их индексами

#include <stdio.h>
 
#include <stdlib.h>
 
#include <locale.h>
 
#include <iostream>
 
#include <clocale>
 
#include <time.h>
 
using namespace std;
 
void fillarr(int arr[], int N)
 
{
 
for (int i = 0; i < N; i++)
 
{
 
arr[i] = rand() % 2;
 
}
 
}
 
void printarr(int arr[], int N1)
 
{
 
for (int i = 0; i < N1; i++)
 
{
 
cout << "arr" << "\t" << arr[i] << endl;
 
}
 
}
 
int f(int* arr, int N)
 
{
 
int j = 0;
 
for (int i = 0; i < N; i++)
 
if (arr[i] == 0)
 
{
 
arr[i] = i; j++;
 
} return j;
 
}
 
void main()
 
{
 
srand(time(0));
 
setlocale(0, "RUS");
 
int N1, N2;
 
cout << "enter arr N1" << endl;
 
cin >> N1;
 
cout << "enter arr N2" << endl;
 
cin >> N2;
 
int* A = new int[N1];
 
int* B = new int[N2];
 
fillarr(A, N1);
 
fillarr(B, N2);
 
cout << "\n\nОригинал:\n" << endl;
 
cout << "\tМассив A : " << endl;
 
printarr(A, N1);
 
cout << "\tМассив B: " << endl;
 
printarr(B, N2);
 
int zamA = f(A, N1 / 2);
 
int zamB = f(B, N2);
 
cout << "\n\nПосле обработки:\n " << endl;
 
cout << "\tМассив A : " << endl;
 
printarr(A, N1);
 
cout << "\t Число замен: " << zamA << endl;
 
cout << "\tМассив B: " << endl;
 
printarr(B, N2);
 
cout << "\t Число замен: " << zamB << endl;
 
delete[] A;
 
delete[] B;
 
system("pause");
 
}

void printarr(int arr[], int N1)
{
    for(int i = 0; i < N1; i++)
        printf("%d", arr[i]);
}

void printarr(int arr[], int N1)
{
    for(int i = 0; i < N1; i++)
        printf("arr[%d] = %d\n", i, arr[i]);
}

// пример обработки - Заменить нулевые значения элементов массива их индексами 
int Work(int* M,int n)
{
    int s=0;
    for(int i=0;i<n;i++)
    {
        if(M[i]==0)
        {
            s++;
            M[i]=i;
    }
    return s; // возвращаем количество замен
}