力扣—2024/4/18—从双倍数组中还原原数组

最新推荐文章于 2024-10-14 18:52:56 发布

嵌入式小小聪

最新推荐文章于 2024-10-14 18:52:56 发布

阅读量255

点赞数 2

文章标签： leetcode

本文链接：https://blog.csdn.net/qq_68915288/article/details/137917524

版权

文章介绍了如何结合快速排序算法和哈希表技术，处理已改变顺序的数组，找到原始数组中未变部分。函数`findOriginalArray`首先对改变后的数组进行排序，然后利用哈希表跟踪元素出现次数，找出未改变位置的元素。

摘要由CSDN通过智能技术生成

代码实现：

快排 + 哈希表 ——超时

/**
 * Note: The returned array must be malloced, assume caller calls free().
 */
void swap(int *m, int *n) {
    int temp = *m;
    *m = *n;
    *n = temp;
}

// 快排
void sort(int *a, int l, int r) { // 左闭右开
    if (r - l <= 2) {
        if (r - l <= 1) {
            return;
        } 
        if (a[l] > a[r - 1]) {
            swap(&a[l], &a[r - 1]);
        }
        return;
    }
    int i = l, j = r - 1;
    int pivot = a[i];
    while (i < j) {
        while (i < j && a[j] >= pivot) {
            j--;
        }
        if (i < j) {
            a[i] = a[j];
            i++;
        }
        while (i < j && a[i] <= pivot) {
            i++;
        }
        if (i < j) {
            a[j] = a[i];
            j--;
        }
    }
    a[i] = pivot;
    sort(a, l, i);
    sort(a, i + 1, r);
}

int* findOriginalArray(int *changed, int changedSize, int *returnSize) {
    *returnSize = 0;
    if (changedSize == 0 || changedSize % 2) {
        return NULL;
    }
    int *res = malloc(sizeof(int) * changedSize / 2);
    sort(changed, 0, changedSize);
    int hash[100001] = {0};
    for (int i = 0; i < changedSize; i++) {
        hash[changed[i]]++;
    }
    for (int i = 0; i < changedSize; i++) {
        if (hash[changed[i]] == 0) {
            continue;
        }
        if (changed[i] == 0 && hash[0] < 2) { // 特别处理0的情况
            *returnSize = 0;
            return NULL;
        }
        if (hash[changed[i] * 2] == 0) {
            *returnSize = 0;
            return res;
        }
        res[(*returnSize)++] = changed[i];
        hash[changed[i]]--;
        hash[changed[i] * 2]--;
    }
    return res;
}