Из одного массива сформировать другой

#include <iostream>
#include <conio.h>
 
#define N 20
 
int main()
{
    int A[N], B[N], i, j, n;
 
    printf("Vvedite kolichestvo elementov: ");
    scanf("%d",&n);
 
    for (i=0; i<n; i++)
        scanf("%d",&A[i]);
 
    for (i=0; i<n; i++) 
        if (i%2)
            B[i]=A[i]+i;
        else B[i]=A[i]-i;
 
    for (i=0; i<n; i++)
        printf("%d ",B[i]);
 
    _getch();
    return 0;
}

#include <iostream>
#include <vector>
 
int main()
  {
  std::cout << "Enter N: ";
  size_t n;
  std::cin >> n;
  std::vector<int> a(n);
  std::vector<int> b(n);
  for (size_t i = 0; i < n; ++i)
    {
    std::cout << "a[" << i << "] = ";
    std::cin >> a[i];
    if (i & 1)
      b[i] = a[i] + i;
    else
      b[i] = a[i] - i;
    }
  for (auto i : b)
    std::cout << i << " ";
  std::cin.get();
  std::cin.get();
  return 0;
  }