题解 | 数组中的逆序对
数组中的逆序对
https://www.nowcoder.com/practice/96bd6684e04a44eb80e6a68efc0ec6c5
归并排序,先分再和
分:在此处递归得分割数组,直到有序,也就是分割到一个数组只有单个元素
mergeSort(nums, left, mid); mergeSort(nums, mid + 1, right);
和:通过 merge(nums, left, mid, right);把分开的数组排序完合回来
到此处我们完成了mergeSort函数,代码如下:
void mergeSort(vector<int>& nums, int left, int right) { if (left < right) { int mid = left + (right - left) / 2; mergeSort(nums, left, mid); mergeSort(nums, mid + 1, right); merge(nums, left, mid, right); } }
接下来就需要完成merge函数,正是本题找逆序对的关键点
在合的过程中,如果前面元素小于后面元素,则不能构成逆序对,如果后面元素小于前面元素,则可以和前面元素中未排序的元素都可以组成逆序对,即ans = (ans+(mid - i + 1)) % MOD;
long long ans = 0; void merge(vector<int>& nums, int left, int mid, int right) { // 此处需要提前分配好空间,否则在21行tmp[k++],由于未初始化导致段错误 vector<int> tmp(right-left+1); int k = 0; int i = left, j = mid + 1; while (i <= mid && j <= right) { if (nums[i] <= nums[j]) // 如果前面元素小于后面元素,则不能构成逆序对 { tmp[k++] = nums[i++]; } else //如果前面元素大于后面元素,可以构成逆序对 { tmp[k++] = nums[j++]; ans = (ans+(mid - i + 1)) % MOD; } } //将剩下元素插入tmp while (i <= mid) { tmp[k++] = nums[i++]; } while (j <= right) { tmp[k++] = nums[j++]; } // 将tmp数组写回 for (int m = 0; m < tmp.size(); m++) { nums[left + m] = tmp[m]; } }
最终ac代码如下:
#include <vector> #define MOD 1000000007 class Solution { public: /** * 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可 * * * @param nums int整型vector * @return int整型 */ long long ans = 0; void merge(vector<int>& nums, int left, int mid, int right) { // 此处需要提前分配好空间,否则在的21行tmp[k++],由于未初始化导致段错误 vector<int> tmp(right-left+1); int k = 0; int i = left, j = mid + 1; while (i <= mid && j <= right) { if (nums[i] <= nums[j]) // 如果前面元素小于后面元素,则不能构成逆序对 { tmp[k++] = nums[i++]; } else //如果前面元素大于后面元素,可以构成逆序对 { tmp[k++] = nums[j++]; ans = (ans+(mid - i + 1)) % MOD; } } while (i <= mid) { tmp[k++] = nums[i++]; } while (j <= right) { tmp[k++] = nums[j++]; } // 将tmp数组写回 for (int m = 0; m < tmp.size(); m++) { nums[left + m] = tmp[m]; } } void mergeSort(vector<int>& nums, int left, int right) { if (left < right) { int mid = left + (right - left) / 2; mergeSort(nums, left, mid); mergeSort(nums, mid + 1, right); merge(nums, left, mid, right); } } int InversePairs(vector<int>& nums) { ans = 0; mergeSort(nums, 0, nums.size()-1); return (int)ans; } };
小菜b写了一些自己的理解,望大佬指正