判断数组中所有的数字是否只出现一次

判断数组中所有的数字是否只出现一次

题目描述

给定一个个数字arr，判断数组arr中是否所有的数字都只出现过一次。

输入描述：

输入包括两行，第一行一个整数$n(1\le n\le 10^5)$，代表数组arr的长度。第二行包括n个整数，代表数组$arr(1\le arr[i]\le 10^7)$。

输出描述：

如果arr中所有数字都只出现一次，输出“YES”，否则输出“NO”。

示例1

输入

3
1 2 3

输出

YES

示例2

输入

3
1 2 1

输出

NO

备注：

要求
1.时间复杂度$O(n)$。
2.额外空间复杂度$O(1)$。

---

题解：

额外空间复杂度和时间复杂度决定了只能使用堆来解决。

但是建堆有两种方式：

• 自顶向下

  时间复杂度为 $O(nlogn)$

• 自下向上

  时间复杂度为$O(n)$

所以就自己手写一个自下向上的堆吧，在向上调整过程中判断是否存在重复元素即可。

注意：判断重复元素可能的情况有很多种，我使用的是小根堆，分了三种情况讨论，但具体是否遗漏了什么，还需要牛逼网友帮忙debug。

1. a[u] == a[u - 1]
2. a[u] == a[u >> 1]
3. a[u] == a[(u + 1) >> 1]

比较有意思的是，若使用大根堆，对于测试数据只需要和父节点判重即可，我这里提供一组 hack 数据：

5
5 4 3 2 2

大根堆代码（有问题）：

#include <cstdio>
#include <algorithm>

const int N = 100010;

using namespace std;

int n;
int a[N];

bool up( int u ) {
    while ( u > 1 ) {
        if ( a[u >> 1] == a[u] ) return false;
        if ( a[u >> 1] < a[u] ) {
            swap( a[u >> 1], a[u] );
            u >>= 1;
        } else break;
    }
    return true;
}

int main(void) {
    scanf("%d", &n);
    int x;
    bool flag = true;
    for ( int i = 1; i <= n; ++i ) {
        scanf("%d", a + i);
        if ( !up( i ) ) {
            flag = false;
            break;
        }
    }
    puts( flag ? "YES" : "NO" );
    return 0;
}

小根堆代码（不确定是否有问题）：

#include <cstdio>
#include <algorithm>

const int N = 100010;

using namespace std;

int n;
int a[N];

bool up( int u ) {
    while ( u > 1 ) {
        if ( a[u] == a[u - 1] ) return false;
        if ( a[u >> 1] == a[u] ) return false;
        if ( a[u] == a[(u + 1) >> 1] ) return false;
        if ( a[u >> 1] > a[u] ) {
            swap( a[u >> 1], a[u] );
            u >>= 1;
        } else break;
    }
    return true;
}

int main(void) {
    scanf("%d", &n);
    int x;
    bool flag = true;
    for ( int i = 1; i <= n; ++i ) {
        scanf("%d", a + i);
        if ( !up( i ) ) {
            flag = false;
            break;
        }
    }
    puts( flag ? "YES" : "NO" );
    return 0;
}