创新工厂笔试题------数组反序个数

来源:互联网 发布:sql2000数据库无法启动 编辑:程序博客网 时间:2024/05/21 22:57

创新工场笔试题:

给定一整型数组,若数组中某个下标值大的元素值小于某个下标值比它小的元素值,称这是一个反序。
即:数组a[]; 对于i < j 且 a[i] > a[j],则称这是一个反序。
给定一个数组,要求写一个函数,计算出这个数组里所有反序的个数。

创新工场的这道求反序个数的题就是算法导论中的题目。其实是对归并排序做些修改,算法的复杂度是nlgn。

程序代码如下:

//算法思想:并归排序数组S#include "stdafx.h"#include <iostream>using namespace std;#include <malloc.h>#include <ctime>int merge(int *a, int low, int mid, int high);int mergesort(int *a, int low, int high);int main(){int i,k,length;srand((unsigned)time(NULL));length=rand()%20;//随即产生数组长度int *arry=new int[length];for (i=0;i<length;i++){arry[i]=rand()%100;//产生伪随机数}cout<<"数组归并排序前: ";//输出初始关键字序列for (k=0;k<length;k++)cout<<arry[k]<<"  ";cout<<endl;int count=mergesort(arry,0,length-1);//归并排序返回逆数对cout<<"数组归并排序后: ";//输出初始关键字序列for (k=0;k<length;k++)cout<<arry[k]<<"  ";cout<<"\n逆序对为:"<<count<<endl;getchar();return 0;}int merge(int *a, int low, int mid, int high){int n1 = mid-low+1;int n2 = high-mid;int *tmp1 =(int*)malloc(n1*sizeof(int));int *tmp2 = (int*)malloc(n2*sizeof(int));int i, j, k;for(i=0; i<n1; i++){tmp1[i] = a[low+i];}for(i=0; i<n2; i++){tmp2[i] = a[mid+i+1];}int count=0;k = low;for(i=0, j=0; i<n1 && j<n2;){if(tmp1[i]<=tmp2[j]){a[k++] = tmp1[i++];}else{a[k++] = tmp2[j++];count+=(n1-i);//计算逆数对}}while(i<n1){a[k++] = tmp1[i++];}while(j<n2){a[k++] = tmp2[j++];}free(tmp1);free(tmp2);return count;}int mergesort(int *a, int low, int high){int count=0;if(low < high){int mid = (low+high)/2;count+=mergesort(a, low, mid);count+=mergesort(a, mid+1, high);count+=merge(a, low, mid, high);}return count;}