Hdu 4422 The Little Girl who Picks Mushrooms

题意：有五座山，已知小女孩已经采过了的山，另外还剩下的（5-n）座山可以去采，出口时必须要把三个袋子之和为1024克的整数倍给Sunny, Lunar and Star，没有就全部留下袋子，女孩手中的蘑菇=0，若有符合的三个袋子，出口后回家的路上被Marisa重复拿走1024克，知道女孩手中剩下不多于1024克的蘑菇，问最终女孩最多能有多少蘑菇

思路：n<=3的时候：另外的山肯定可以使最后回去的时候达到最大值1024

          n==4的时候：枚举三个可以凑成1024的整数倍，否则，就拿另外一个山来补

          n==5的时候：全部枚举

代码：

#include <iostream>
#include <stdio.h>
#include <cmath>
#include <cstring>
#include <algorithm>
#include <vector>
#include <map>
using namespace std;
const int M=1e5+5;
const int mod=1e9+7;
int a[6];
int main()
{
    int n;
    while(scanf("%d",&n)!=EOF)
    {
        for(int i=1;i<=n;i++)
        scanf("%d",&a[i]);
        int sum=0,ans;
        if(n<=3)
        {
            printf("1024\n");
            continue;
        }
        if(n==4)
        {
            ans=0;
            for(int i=1;i<=4;i++)
            {
                for(int j=1;j<=4;j++)
                {
                    for(int k=1;k<=4;k++)
                    {
                        if(i!=j && j!=k && i!=k)
                        {
                            int t=a[i]+a[j]+a[k];
                            if(t%1024==0)
                            {
                                ans=1024;
                                i=5,j=5,k=5;
                            }
                        }
                    }
                }
            }
            if(ans==1024)
            {
                printf("1024\n");
                continue;
            }
            for(int i=1;i<=4;i++)
            {
                for(int j=1;j<=4;j++)
                {
                    if(i!=j)
                    {
                        int t=0;
                        for(int k=1;k<=4;k++)
                        {
                            if(k!=i && k!=j)
                            {
                                t+=a[k];
                            }
                        }
                        while(t>1024)
                        t-=1024;
                        if(t>sum)
                        sum=t;
                    }
                }
            }
            printf("%d\n",sum);
            continue;
        }
        if(n==5)
        {
            int ff=0;
            for(int i=1;i<=5;i++)
            {
                for(int j=1;j<=5;j++)
                {
                    for(int k=1;k<=5;k++)
                    {
                        if(i==j || i==k || j==k)continue;
                        int t=a[i]+a[j]+a[k];
                        ans=0;
                        if(t%1024==0)
                        {
                            ff=1;
                            for(int x=1;x<=5;x++)
                            {
                                if(x!=i && x!=j && x!=k)
                                {
                                    ans+=a[x];
                                }
                            }
                            while(ans>1024)
                            ans-=1024;
                            if(ans>sum)
                            sum=ans;
                        }
                    }
                }
            }
            if(ff==0)printf("0\n");
            else
            printf("%d\n",sum);
        }
    }
    return 0;
}