HDOJ搜索专题之Sticks

题目数学模型：给定n个正整数，现要将这n个数分成k组，且满足每组的和都相等。求最多能分多少组。

这题是经典的剪枝搜索题，原题来自PKU。下面的程序虽然AC了，但是跑不动POJ中discuss中的那组BT数据。

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <vector>
#define MAX(a,b) ((a)>(b)?(a):(b))
#define N 64
using namespace std;
vector<int>stick[N];
int len[N],vis[N],n,sum,yes;
int cmp(const void *a,const void *b)
{
  return *(int*)b-*(int*)a;
}
void dfs(int k,int l,int L)
{
  int i;
  if(l==L)
  {
    dfs(k+1,0,L);
    return;
  }
  if(k==sum/L && l==0)
  {
    yes=1;
    return;
  }
  if(l==0)
  {
    if(k==0)  i=0;
    else  i=stick[k-1].front()+1;
    for(;i<n && vis[i];i++);
    if(i<n && len[i]<=L)
    {
      vis[i]=1;
      stick[k].push_back(i);
      dfs(k,len[i],L);
      stick[k].pop_back();
      vis[i]=0;
    }
  }
  else
  {
    for(i=stick[k].back()+1;!yes && i<n;i++)
    {
      if(l+len[i]>L || vis[i])  continue;
      vis[i]=1;
      stick[k].push_back(i);
      dfs(k,l+len[i],L);
      stick[k].pop_back();
      vis[i]=0;
      if(l+len[i]==L) break;
    }
  }
}
int main()
{
  int i,max;
  while(scanf("%d",&n)&&n)
  {
    sum=max=0;
    for(i=0;i<n;i++)
    {
      scanf("%d",&len[i]);
      sum+=len[i];
      max=MAX(max,len[i]);
    }
    qsort(len,n,sizeof(len[0]),cmp);
    yes=0;
    for(i=max;i<=sum;i++)
    {
      if(sum%i) continue;
      for(int j=0;j<n;j++)  stick[j].clear();
      memset(vis,0,sizeof(vis));
      dfs(0,0,i);
      if(yes) break;
    }
    printf("%d\n",i);
  }
  return 0;
}

BT数据如下：

64
40 40 30 35 35 26 15 40 40 40 40 40 40 40 40 40 40 40 40 40 40 40 40 43 42 42 41 10 4 40 40 40 40 40 40 40 40 40 40 40 40 40 40 25 39 46 40 10 4 40 40 37 18 17 16 15 40 40 40 40 40 40 40 40

0