交换两个数组使两个数组和的差最小

转自   南多的技术剥壳:http://www.cnblogs.com/nanduo/archive/2009/06/29/1513035.html 


有两个序列a,b,大小都为n,序列元素的值任意整数,无序;
要求:通过交换a,b 中的元素,使[序列a 元素的和]与[序列b 元素的和]之间的差最小。
例如:
var a=[100,99,98,1,2, 3];
var b=[1, 2, 3, 4,5,40];


求解思路:

    当前数组a和数组b的和之差为
    A = sum(a) - sum(b)

    a的第i个元素和b的第j个元素交换后,a和b的和之差为
    A' = sum(a) - a[i] + b[j] - (sum(b) - b[j] + a[i])
           = sum(a) - sum(b) - 2 (a[i] - b[j])
           = A - 2 (a[i] - b[j])
    设x = a[i] - b[j]

    |A| - |A'| = |A| - |A-2x|

    假设A > 0,

    当x 在 (0,A)之间时,做这样的交换才能使得交换后的a和b的和之差变小,x越接近A/2效果越好,

    如果找不到在(0,A)之间的x,则当前的a和b就是答案。

    所以算法大概如下:

    在a和b中寻找使得x在(0,A)之间并且最接近A/2的i和j,交换相应的i和j元素,重新计算A后,重复前面的步骤直至找不到(0,A)之间的x为止。


int test(float a[], float b[], int n)
 {
     float sumA, sumB; //sumA为数组a总和,sumB为数组b总和
     float sum_diff, num_diff; //sum_diff为a,b总和差, num_diff为a,b中各选的两个数之差
     float temp1, temp2;    //temp1为 每轮sum_diff/2, temp2为每轮所选两个数之差于temp1最接近的那个
     int i, j;
     float temp; //用于对调a,b间数
     int tempi, tempj;    //每轮所选两个数之差于temp1最接近的那组数
     unsigned int flag_sum = 0, flag_num = 0;  //flag_sum为1, sumA大于sumB; flag_num为1, 此轮存在两个数之差小于sum_diff
 
 
         
 
     while(1){
 
         //算出a,b数组和
         sumA = 0;
         sumB = 0;
         for(i=0;i < n;i++)
         {
             sumA += a[i];
             sumB += b[i];
         }
 
         if(sumA >= sumB){
             sum_diff = sumA - sumB;
             flag_sum = 1;
         }
         else
             sum_diff = sumB - sumA;    
     
         temp1 = sum_diff/2;
         temp2 = temp1;
         tempi = 0;
         tempj = 0;    
     
         //找出a,b间差值最接近sum_diff/2的那一对数
         if(flag_sum == 1){    //sumA > sumB
             for(i=0; i < n; i++)
                 for(j=0; j < n; j++)
                 
                     if(a[i] > b[j]){
                         num_diff = a[i] - b[j];
                         if(num_diff < sum_diff){
                             flag_num =1;
                             if(num_diff >= temp1){
                                 if(num_diff-temp1 < temp2){
                                     temp2 = num_diff-temp1;
                                     tempi = i;
                                     tempj = j;
                                 }
                             }
                             else{
                                 if(temp1 - num_diff < temp2){
                                     temp2 = temp1 - num_diff;
                                     tempi = i;
                                     tempj = j;
                                 }
                             }
                         }
                     }
         }
         else{
             for(i=0; i < n; i++)
                 for(j=0; j < n; j++)
                 
                     if(a[i] < b[j]){
                         num_diff = b[j] - a[i];
                         if(num_diff < sum_diff){
                             flag_num =1;
                             if(num_diff >= temp1){
                                 if(num_diff-temp1 < temp2){
                                     temp2 = num_diff-temp1;
                                     tempi = i;
                                     tempj = j;
                                 }
                             }
                             else{
                                 if(temp1 - num_diff < temp2){
                                     temp2 = temp1 - num_diff;
                                     tempi = i;
                                     tempj = j;
                                 }
                             }
                         }
                     }
         }
 
         if(flag_num == 0)
             break;
 
         temp = a[tempi];
         a[tempi] = b[tempj];
         b[tempj] = temp;
     
         flag_num = 0;
         flag_sum = 0;
     }
         
     for(i=0; i < n;i++)
         printf("%f\t",a[i]);
 
     printf("\n");
 
     for(i=0; i < n;i++)
         printf("%f\t",b[i]);
 
     printf("\n");    
     
     return 0;
 }
 
 
 int main(int argc, char *argv[])
 {
 
     float a[3] = {4,5,12};
     float b[3] = {1,2,3};
 
     test(a, b, 3);
 
     return 0;
 }


评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值