2354339_ac_15ms_60k.c

来自「北大大牛代码 1240道题的原代码 超级权威」· C语言 代码 · 共 29 行

#include <stdio.h>

int power(int a,int b)
{
	int i, tmp = 1;
	
	for(i = 1; i <= b; i++)
		tmp *= a;
	return tmp;
}

int main()
{
	int i, a;
	int n, k, ans = 0;

	scanf("%d%d",&n,&k);
	for(i = 0; i < n; i++)
	{
		scanf("%d",&a);
		if(a>0)
			ans += power(a,k);
		else
			if(k==2)
				ans += a*a;
	}
	printf("%d",ans);
	return 1;
}