关于集合问题

zwh1481017739

例如  集合a  集合 b  求b在a中的余集
集合a   1 2 3 4 5 6
集合b    2 4 5 6 7 8
输出   1 3

人造人

1. #include <iostream>
   
2. #include <vector>
   
3. 
   
4. const std::vector<int> complement(const std::vector<int> &a, const std::vector<int> &b)
   
5. {
   
6.         std::vector<int> res;
   
7.         for(const auto i: a)
   
8.         {
   
9.                 bool flag = true;
   
10.                 for(const auto j: b)
    
11.                 {
    
12.                         if(i == j)
    
13.                         {
    
14.                                 flag = false;
    
15.                                 break;
    
16.                         }
    
17. 
    
18.                 }
    
19.                 if(flag)
    
20.                         res.push_back(i);
    
21.         }
    
22.         return res;
    
23. }
    
24. 
    
25. std::ostream &operator<<(std::ostream &os, const std::vector<int> &data)
    
26. {
    
27.         for(const auto i: data)
    
28.         {
    
29.                 std::cout << i << " ";
    
30.         }
    
31.         return os;
    
32. }
    
33. 
    
34. int main()
    
35. {
    
36.         std::vector<int> a = {1, 2, 3, 4, 5, 6};
    
37.         std::vector<int> b = {2, 4, 5, 6, 7, 8};
    
38.         std::cout << complement(a, b) << std::endl;
    
39.         return 0;
    
40. }
    

复制代码

荣耀

a = [1,2,3,4,5,6]
b = [2,4,5,6,7,8]
for i in a:
        if i not in b:
                print(i,end=" ")
楼上写的太高深看不到

人造人

荣耀 发表于 2019-12-21 15:33
a = [1,2,3,4,5,6]
b = [2,4,5,6,7,8]
for i in a:

这里是C/C++，不是py

Croper

提供一个排序之后的算法，效率要高一些：

1. #include <iostream>
   
2. #include <vector>
   
3. #include <algorithm>
   
4. using namespace std;
   
5. 
   
6. template <typename T>
   
7. vector<T> operator-(vector<T> a, vector<T> b) {
   
8.         sort(a.begin(), a.end());
   
9.         sort(b.begin(), b.end());
   
10.         vector<T> ret;
    
11.         for (auto p = a.begin(), q = b.begin(); p != a.end(); ++p) {
    
12.                 while (q != b.end() && *p > *q) {
    
13.                         q++;
    
14.                 }
    
15.                 if (q != b.end() && *p == *q) {
    
16.                         continue;
    
17.                 }
    
18.                 ret.push_back(*p);
    
19.         }
    
20.         return ret;
    
21. }
    
22. 
    
23. template <typename T>
    
24. ostream& operator<<(ostream& os, vector<T> v) {
    
25.         os << "{ ";
    
26.         bool b = false;
    
27.         for (const T& t : v) {
    
28.                 if (b) {
    
29.                         os << " , ";
    
30.                 }
    
31.                 os << t;
    
32.                 b = true;
    
33.         }
    
34.         os << " }";
    
35.         return os;
    
36. }
    
37. int main() {
    
38.         vector<int> a = { 1,2,3,4,5,6 };
    
39.         vector<int> b = { 2,4,5,6,7,8 };
    
40.         vector<int> c = a - b;
    
41.         cout << c << endl;
    
42.         system("pause");
    
43. }

复制代码