编程之美-数组分割

 

编程之美'数组分割'和扩展(扩展只需要转化为整数数组,加个固定的偏移)

将一个数组划分成两个子数组,要求他们的和最接近
1.长度要求相等的情况: 
2.长度没有要求的情况:

都能用动态规划解决


#include
#include
using namespace std;


void ArraySplit1(int a[], int n)//两个子数组长度要求相等
{
int dp[20][20][200]; //k:当前元素   i:元素个数  j:元素和
int sum=0;
for(int k=0; k
sum += a[k];
cout<<"sum: "<
 
for(int k=0; k<=n; k++) 
dp[k][0][0] = 0;
for(int k=1; k<=n; k++)
for(int i=0; i<=min(k,n/2); i++)
for(int j=0; j<=sum/2; j++) //
{
if(a[k-1]<=j && dp[k-1][i-1][j-a[k-1]] >= 0)
dp[k][i][j] = k;  
else if(k>i)
dp[k][i][j] = dp[k-1][i][j];
else
dp[k][i][j] = -1;  
}
int k=n;
int i=n/2;
int j=sum/2;
while(dp[k][i][j] <0 )
j--;
int part = 0;
while(i>0)
{
if(dp[k][i][j] == k)
{
cout<
part += a[k-1];
j -= a[k-1];
k--;
i--;
}
else if(dp[k][i][j] > 0)
{
--k;
}
}
cout<
cout<<"part: "<
        
}



void ArraySplit2(int a[], int n) //两个子数组长度不要求相等
{
int dp[20][200];
int sum=0;
for(int k=1; k<=n; k++)
{
sum += a[k-1];
dp[k][0] = 0;
dp[0][k] = -1;
}
dp[0][0] = 0;
cout<<"sum: "<
for(int k=1; k<=n; k++)
{
for(int j=1; j<=sum/2; j++)
{
if(j>=a[k-1] && dp[k-1][j-a[k-1]]>=0)
dp[k][j] = k;
else
dp[k][j] = dp[k-1][j];
}
}
int j=sum/2;
int k=n;
while(dp[k][j] < 0)
j--;
int part=0;
while(k>0)
{
if(dp[k][j] == k)
{
cout<
part += a[k-1];
j -= a[k-1];
k--;
}
else
{
k--;
}
}
cout<
cout<<"part: "<
}




int main()
{
int a[] = {1,5,7,8,9,6,3,15,21,18,30,23};
ArraySplit1(a,12); 
ArraySplit2(a,12); 
getchar();

return 0;
}

你可能感兴趣的:(算法,动态规划)