POJ 2209 The King#贪心

Posted 2020-06-17

(～￣▽￣)～*  水

//水题；潜力^e为正数(e为2时都可以)的儿子都可以去上战场了，
//英文要看懂，exponent指数，不超过3的正数
#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;

int main()
{
    int n,e,son[105];
    while(~scanf("%d",&n))
    {
        int cha=0;
        scanf("%d",&e);
        for(int i=0;i<n;i++)
            scanf("%d",&son[i]);
        if(e==3)
        {
            for(int i=0;i<n;i++)
                if(son[i]>0)
                    cha+=son[i]*son[i]*son[i];
        }
        else if(e==2)//注意e为2时，都为平方后都为正数
        {
            for(int i=0;i<n;i++)
                    cha+=son[i]*son[i];
        }
        else if(e==1)
        {
            for(int i=0;i<n;i++)
                if(son[i]>0)
                    cha+=son[i];
        }
        printf("%d\n",cha);
    }
    return 0;
}