首页 > 代码库 > 量数组交换差最小算法

量数组交换差最小算法

 

交换两个数组值使两个数组之差最小

有两个序列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中元素的和为sum_a和sum_b。假设aa和bb分别为序列a,b中的元素,则交换aa,bb后序列的和变为sum_a-aa+bb,sum_b+aa-bb;

两序列的差为(sum_a-aa+bb)-(sum_b+aa-bb)=sum_a-sum_b-2*(aa-bb);

调换一次,差值前后变化为2*(aa-bb)

所以可以扫描序列a,b中的元素,找到使abs(sum_a-sum_b-2*(aa-bb))最小的两个元素进行交换,重复此过程,直至两序列的差无法减小。

bool Swap2Balance(int *pa, int *pb, int n)

 {

  int suma=0,sumb=0;

  for (int i=0;i<n;i++)

  {

   suma+=pa[i];

   sumb+=pb[i];

  }

  int diff=suma-sumb;

  while (diff!=0)

  {

   int besti=0,bestj=0;

   int bestchange=0;

   for(int i=0;i<n;i++)

    for (int j=0;j<n;j++)

    {

     int change=(pa[i]-pb[j]);

     //交换后差为(suma-pa[i]+pb[j])-(sumb+pa[i]-pb[j])=diff-2*change

     if (abs(diff-2*change)<abs(diff-2*bestchange))

     {

      bestchange=change;

      besti=i;

      bestj=j;

     }

    }

   if (bestchange==0) //差不能再缩小

   return false;

   int temp=pa[besti];

   pa[besti]=pb[bestj];

   pb[bestj]=temp;

   suma-=bestchange;

   sumb+=bestchange;

   diff=suma-sumb;

  }

  return true;

 }

两重循环, 计算复杂度 O(n^2)

量数组交换差最小算法