XDOJ705.数组折叠

#define _CRT_SECURE_NO_WARNINGS

#include<stdio.h>
#include<math.h>

void fold(int n,int count, int arr[])
{
	while (count)
	{
		for (int i = 0; i < pow(2, n - 1); i++)
		{
			arr[i] = arr[i] + arr[(int)pow(2, n) - 1 - i];
		}
		n--;
		count--;
	}
	return;
}

int main()
{
	int n = 0;
	int m = 0;
	scanf("%d %d", &n, &m);
	int arr[300] = { 0 };
	for (int i = 0; i < pow(2, n); i++)
	{
		scanf("%d", &arr[i]);
	}
	fold(n,m, arr);
	for (int i = 0; i < pow(2, n - m); i++)
	{
		printf("%d ", arr[i]);
	}
	return 0;
}