面试题系列-整数数组长度为n,分为m份使各份的和相等,求最大的m

来源:互联网 发布:bool c语言 编辑:程序博客网 时间:2024/04/28 07:51

这应该是网上比较常见的题目了,但由于本人新手,决定练练手,实现一下。

解题思路:注意m,肯定能被数组的和整除,得到的结果是分成m份之后的和。可以考虑从m=n时开始递归,直接求得符合条件的m即为最大的,代码如下:

#include <iostream>using namespace std;int findsubsum(int a[],int n,int m,int goal,int aux[],int sum,int groupid){if(goal <0)return 0;if(goal ==0){groupid++;goal = sum;if(groupid == m+1)return 1;}for(int i=0;i<n;i++){if(aux[i] != 0)continue;aux[i] = groupid;if(findsubsum(a,n,m,goal-a[i],aux,sum,groupid))return 1;aux[i] =0;}return 0;}int FindMax(int a[],int n){int m;int sum =0;for(int i=0;i<n;i++){sum +=a[i];}int *aux = (int *)malloc(sizeof(int)*n);for(m =n;m>=2;m--){if(sum%m !=0 ) continue;memset(aux,0,sizeof(int)*n);if(findsubsum(a,n,m,sum/m,aux,sum/m,1)){free(aux);return m;}}return 1;}int main(){int m;int a[] = {3,2,4,3,6};m = FindMax(a,5);cout<<m<<endl;system("PAUSE");return 1;}
测试结果是3,正确。但刚开始时把
memset(aux,0,sizeof(int)*n);
写成

memset(aux,0,n);
导致结果错误,看来基础还有待加强