数组小和

题目

思路

数组小和，以及其他相关两数对之间关系的题目，都可以通过这种思考方式进行思考。

通过改写归并排序进行计算，

每次对排序的同时计算小和，

因为每次计算子排序的时候，都能够进行计算一次小和，

当要将两个排好序的两个子序列进行合并时， 当左边的当前值比右边的当前值要小，说明，这个

值比右边的所有 值都要小，所有就产生了arr[i]*(r2-l2+1)个小和，通过这种进行计算，直到计算完全

代码

#include<iostream>
#include<cstdlib>
#include<ctime>
#include<stack>
using namespace std;


void printArr(int N);
void scanfArr(int N);
int arr[1000];

const int INF = 0x3f3f3f3f;

int merge(int l1,int r1,int l2,int r2);

int func(int l, int r){
	if (l == r) return 0;
	if(l+1 == r){
		if(arr[l]<arr[r])
			return arr[l];
		else{
			int temp =arr[l];
			arr[l] = arr[r];
			arr[r] = temp;	
		}
		return 0;
	}
	
	int mid = (l+r)/2;
	int a = func(l,mid);
	int b = func(mid+1,r);
	
	int c = merge(l,mid,mid+1,r); 
	
	return a+b+c;
}

int merge(int l1,int r1,int l2,int r2){
	int start = l1; 
	int *p = new int[r2 - l1+1];	
	int sum = 0;
	int p_indx = 0;
	while(l1<=r1 && l2<=r2){
		if(arr[l1]< arr[l2]){
			p[p_indx] = arr[l1];
			sum+=arr[l1]*(r2-l2+1);
			l1++;
		}else{
			p[p_indx] = arr[l2];
			l2++;
		}
		p_indx++;
	}
	
	while(l1<=r1){
		p[p_indx++] = arr[l1++];
	}
	while(l2<=r2){
		p[p_indx++] = arr[l2++];
	}
	
	for(int i=0;i<p_indx;i++){
		arr[start+i] = p[i];
	}
	delete []p;
	
	return sum;
}

int main(){
	
	int N = 6; 
	scanfArr(N);
	printArr(N);
	
	int sum = func(0, N-1);
	
	printf("%d\n",sum);
	printArr(N);

	return 0;
} 

void randArr(int N){
	
	srand((int)time(0));
	for(int i=0;i<N;i++){
		arr[i] =rand()%11-5; //[-5,5];  
	}
}

void printArr(int N){
	for(int i=0;i<N;i++){
		printf("%d ",arr[i]);
	}
	printf("\n");

}

void scanfArr(int N){
	for(int i=0;i<N;i++){
		scanf("%d",&arr[i]);
	}
}