洛谷P1978 集合

题目描述

集合是数学中的一个概念，用通俗的话来讲就是：一大堆数在一起就构成了集合。集合有如

下的特性：

•无序性：任一个集合中，每个元素的地位都是相同的，元素之间是无序的。

•互异性：一个集合中，任何两个元素都认为是不相同的，即每个元素只能出现一次。

•确定性：给定一个集合，任给一个元素，该元素或者属于或者不属于该集合，二者必居

其一，不允许有模棱两可的情况出现。

例如 A = {1, 2, 3} 就是一个集合。我们可以知道， 1 属于 A ，即 1 ∈ A ； 4 不属于 A ，

即 4 ∉ A 。一个集合的大小，就是其中元素的个数。

现在定义一个特殊的 k-集合，要求满足：

•集合的所有特性

•对任意一个该集合内的元素 x ，不存在一个数 y ，使得 y = kx 并且 y 属于该集合。即

集合中的任意一个数，它乘以 k 之后的数都不在这个集合内

给你一个由 n 个不同的数组成的集合，请你从这个集合中找出一个最大的 k-集合。

输入格式

第 1 行：两个整数： n 和 k

第 2 行：n 个整数： a[i] 表示给定的集合

输出格式

第 1 行：一个整数： ans 表示最大的 k-集合的大小

输入输出样例

输入 #1

6 2	
2 3 6 5 4 10

输出 #1

3

说明/提示

提示：在样例所给集合中，找出的最大的 2-集合为 {4, 5, 6}

•对于 30% 的数据： n, k ≤ 100

•对于 40% 的数据： a[i] ≤ 2^31-1

•对于 70% 的数据： n, k ≤ 5000

•对于 100% 的数据： n, k ≤ 10^5, a[i] ≤ 2^63-1

解题思路（如果有对集合set使用方法不太明确的，可以看一下这篇文章）

这道题的精髓其实就是这句话：“集合中的任意一个数，它乘以 k 之后的数都不在这个集合内。”

我们先用集合中的count函数求出每个数的第k倍有没有在集合中出现，如果有，那就将其删除。最后输出集合的大小就可以了。

最后要注意本题的数据范围为2^63-1，所以要使用long long。

代码实现

#include <iostream>
#include <set>
#include <algorithm>
using namespace std;
const int N = 1e5 + 2;
long long n, k;
long long a[N];
set<long long> s;
set<long long>::iterator it;
int main(){
	cin >> n >> k;
	for(int i = 1; i <= n; i++){
		cin >> a[i];
		s.insert(a[i]);
	}
	for(it = s.begin(); it != s.end(); it++){
		if(s.count((*it) * k) == 1){
			s.erase((*it) * k);
		}
	}
	cout << s.size();
	return 0;
}