HDU 4422 The Little Girl who Picks Mushrooms（简单题）

题目链接：http://acm.hdu.edu.cn/showproblem.php?pid=4422

题目大意：小姑娘背着5个包去山上采蘑菇，每座山上只能用一个背包采集。三个小精灵会要她3个背包，其里面蘑菇的数量必须是1024的倍数，否则5个背包都会被拿走。然后魔术师会从她剩下的背包里取走蘑菇，每次取1024克，直到蘑菇不超过1024克为止。求小姑娘最多能得到多少蘑菇。

Sample Input

1

9

4

512  512  512 512

5

100 200 300 400 500

5

208 308 508 708 1108

 

Sample Output

1024

1024

792

代码如下：

# include<stdio.h>
# include<algorithm>
# include<iostream>
using namespace std;

int n,a[6],sum,ans;

int calc(int x)
{
    while(x > 1024)
        x -= 1024;
    return x;
}

void solve()
{
    for(int i=0; i<n; i++)
        for(int j=i+1; j<n; j++)
            if(n==4  || (sum-a[i]-a[j])%1024==0)
                ans = max(ans,calc(a[i]+a[j]));
}

int main()
{
    while(scanf("%d",&n)!=EOF)
    {
        sum = 0,ans = 0;
        for(int i=0; i<n; i++)
        {
            scanf("%d",&a[i]);
            sum += a[i];
        }
        if(n<=3)
            ans = 1024;
        else if(n==4)
            for(int i=0; i<4; i++)
                if((sum-a[i])%1024==0)
                    ans = 1024;
        if(ans!=1024)
            solve();
        printf("%d\n",ans);
    }
    return 0;
}