整个二维向量中去除完全相同的行

admin1

1. #include <vector>
   
2. #include <unordered_set>
   
3. #include <algorithm>
   
4. 
   
5. struct VectorHash {
   
6.     size_t operator()(const std::vector<double>& vec) const {
   
7.         size_t hash = 0;
   
8.         for (double val : vec) {
   
9.             hash ^= std::hash<double>{}(val) + 0x9e3779b9 + (hash << 6) + (hash >> 2);
   
10.         }
    
11.         return hash;
    
12.     }
    
13. };
    
14. 
    
15. bool equalVectors(const std::vector<double>& v1, const std::vector<double>& v2) {
    
16.     if (v1.size() != v2.size()) return false;
    
17.     std::sort(v1.begin(), v1.end());
    
18.     std::sort(v2.begin(), v2.end());
    
19.     return v1 == v2;
    
20. }
    
21. 
    
22. std::vector<std::vector<double>> removeDuplicateRows(const std::vector<std::vector<double>>& matrix) {
    
23.     std::unordered_set<std::vector<double>, VectorHash, decltype(equalVectors)*> uniqueRows(&equalVectors);
    
24.     std::vector<std::vector<double>> result;
    
25. 
    
26.     for (const auto& row : matrix) {
    
27.         if (uniqueRows.find(row) == uniqueRows.end()) {
    
28.             // 如果当前行不在哈希集合中，则添加到结果向量和哈希集合中
    
29.             result.push_back(row);
    
30.             uniqueRows.insert(row);
    
31.         }
    
32.     }
    
33. 
    
34.     return result;
    
35. }
    
36. 
    
37. int main() {
    
38.     std::vector<std::vector<double>> matrix = {{1.0, 2.0, 2.0, 3.0},
    
39.                                               {1.0, 2.0, 3.0, 1.0},
    
40.                                               {3.0, 2.0, 2.0, 1.0},
    
41.                                               {4.0, 4.0, 5.0},
    
42.                                               {6.0, 6.0, 7.0, 7.0, 7.0}};
    
43. 
    
44.     std::vector<std::vector<double>> result = removeDuplicateRows(matrix);
    
45. 
    
46.     // 输出结果
    
47.     for (const auto& row : result) {
    
48.         for (double val : row) {
    
49.             std::cout << val << " ";
    
50.         }
    
51.         std::cout << std::endl;
    
52.     }
    
53. 
    
54.     return 0;
    
55. }

复制代码