Hoj 3040 Team Mate

题目链接：http://acm.hit.edu.cn/hoj/problem/view?id=3040

题意：给定一串长度为n的数，求任意几个数的组合和能否是n的倍数。只要求输出其中一种情况即可。

这题正解应该是用抽屉原理：依次求出前缀和s[i]，那么前缀和相等的s[i]和s[j]之间的数的和就是n的倍数。即(i,j]

#include <iostream>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <math.h>
#include <map>
#include <queue>
#include <algorithm>
using namespace std;

#define Maxn 10005
int A[Maxn];
int S[Maxn];

int main()
{
   #ifndef ONLINE_JUDGE
      freopen("in.txt","r",stdin);
   #endif
   int n;
   int flag = 0;
   while(scanf(" %d",&n)!=EOF)
   {
      flag = 0;
      memset(S,0,sizeof(S));
      for(int i=1;i<=n;i++)
      {
         scanf(" %d",&A[i]);
         S[i] = (S[i-1] + A[i])%n;
      }
      for(int i=0;i<=n;i++)
      {
         for(int j=i+1;j<=n;j++)
         {
            if(S[i] == S[j])
            {
               flag = 1;
               printf("%d\n",j-i);
               for(int k=i+1;k<=j;k++)
               {
                  if(k == j) printf("%d\n",A[k]);
                  else printf("%d ",A[k]);
               }
               break;
            }
         }
         if(flag) break;
      }
      if(!flag) puts("0");
   }
   return 0;
}

刚开始是用DP求解的，dp[i][j]代表前i个数能否达到和模n等于j。能为True，否则为False.

然后dir[i][j]代表达到模n=j的状态是否需要第i个数。需要为1，否则为0.

其中dp数组可以用滚动数组优化，但是dir好像只能用二维来做，因为记录前驱的状态有两个属性。这样就会MLE.有能优化的朋友教我一下吧。

代码思路应该是没问题吧：

#include <iostream>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <math.h>
#include <map>
#include <queue>
#include <algorithm>
using namespace std;

#define Maxn 10005

int n;
int A[Maxn];
int B[Maxn];

int dir[Maxn][Maxn];
bool dp[Maxn][Maxn];
vector<int> path;

void getPath(int i,int j)
{
   if(i == 0) return;
   if(dir[i][j] == 1) 
   {
      int bState = (j - B[i] + n) % n;
      getPath(i-1,bState);
      path.push_back(A[i]);
   }
   else getPath(i-1,j);
}
void printPath()
{
   printf("%d\n",path.size());
   for(int i=0;i<path.size();i++)
   {
      if(i == path.size()-1) printf("%d\n",path[i]);
      else printf("%d ",path[i]);
   }
}
int main()
{
   #ifndef ONLINE_JUDGE
      freopen("in.txt","r",stdin);
   #endif
   
   while(scanf(" %d",&n)!=EOF)
   {
      int flag = 0;
      for(int i=1;i<=n;i++)
      {
         scanf(" %d",&A[i]);
         B[i] = A[i]%n;
      }

      memset(dp,false,sizeof(dp));
      memset(dir,0,sizeof(dir));
      path.clear();

      dp[0][0] = true;

      for(int i=1;i<=n;i++)
      {
         for(int j=0;j<n;j++)
         {
            int bState = (j - B[i] + n) % n;
            
            if(i == 1) dp[i][j] = dp[i-1][bState];
            else dp[i][j] = dp[i-1][j] | dp[i-1][bState];
            
            if(dp[i][j])
            {
               //必须优先选择这个，因为有i==1的情况
               if(dp[i-1][bState]) dir[i][j] = 1;
               else dir[i][j] = 0;
               if(j == 0)
               {
                  flag = 1;
                  getPath(i,j);
                  printPath();
                  break;
               }
            }
         }
         if(flag) break;
      }
   }
   return 0;
}