Массив

[An90]

Дан массив А размера N. Сформировать новый массив В того же размера, элементы которого определяются следующим образом: Вк=2*Ак, если Ак<5, Ак/2 в противном случае.
Язык программирования-СИ.
Текст программы:
#ifdef USE_ARRAY
# include <stdlib.h>

int main() {
unsigned i, n;
int* a;
int* b;
scanf("%u", &n);
a = (int*)malloc(n * sizeof(int));
b = (int*)malloc(n * sizeof(int));
for (i = 0; i < n; i++) {
scanf("%d", a + i);
printf("%d", b[i] = a[i] < 5? a[i] + a[i]: a[i] >> 1);
}
free(a);
free(b);
}

#else

int main() {
unsigned n;
int temp;
scanf("%u", &n);
while (n--) {
scanf("%d", &temp);
printf("%d", temp < 5? temp + temp: temp >> 1);
}
}

#endif
Нельзя ли написать как-нибудь по-проще?

[Son Of Pain]

Можно )

#include <algorithm>
#include <iterator>
#include <iostream>
#include <vector>
using namespace std;
int main()
{
	cout << "Enter array size: ";
	vector<int> a(*istream_iterator<size_t>(cin)), b;
	cout << "And elements: ";
	generate(a.begin(), a.end(), []() { return *istream_iterator<int>(cin); } );
	transform(a.begin(), a.end(), back_inserter(b), [](int val) { return (val < 5 ? val * 2 : val / 2); } );
	copy(b.begin(), b.end(), ostream_iterator<int>(cout, " "));
}

[An90]

Спасибо!!! А можно на Си написать и с использованием массива?)