POJ 2209 The King#贪心

(～￣▽￣)～*  水

//水题；潜力^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;

}