boss_face_dir_data_vec.reserve(boss_faces.size()); std::unordered_map<Vector3d, int> boss_faces_dirs_map; for (int b = 0; b < boss_faces_dirs.size(); b++) { boss_faces_dirs_map[boss_faces_dirs[b]] = b + 1; } for (int a = 0; a < boss_faces.size(); a++) { double temp_dir1[3]; getFaceDir(boss_faces[a], temp_dir1); Vector3d temaa(temp_dir1[0], temp_dir1[1], temp_dir1[2]); auto it = std::find_if(boss_faces_dirs_map.begin(), boss_faces_dirs_map.end(),temaa){ return isEqual(temaa, temaa); }); BossFacesDirData boss_face_dir_data; boss_face_dir_data.dir_num = (it != boss_faces_dirs_map.end()) ? it->second : 0; std::copy_n(temp_dir1, 3, boss_face_dir_data.face_dir); boss_face_dir_data_vec.emplace_back(std::move(boss_face_dir_data)); if (boss_face_dir_data.dir_num > 20) { break; } } 改进这段代码
时间: 2024-04-17 11:28:07 浏览: 99
根据你提供的代码,我对其进行了一些改进。主要的改进包括:
1. 使用 `reserve` 方法为 `boss_face_dir_data_vec` 预分配足够的空间,避免不必要的内存重新分配。
2. 将 `boss_faces_dirs_map` 的键类型改为 `Vector3d`,以便与 `boss_faces_dirs` 中的元素类型匹配。
3. 在 `std::find_if` 中使用 lambda 表达式来进行查找,并修正了 `isEqual` 函数的调用。
4. 修正了 `temaa` 的使用,确保在 `std::find_if` 中正确地使用它。
下面是改进后的代码:
```cpp
#include <iostream>
#include <vector>
#include <unordered_map>
#include <algorithm>
struct BossFacesDirData {
int dir_num;
double face_dir[3];
};
bool isEqual(const Vector3d& dir1, const Vector3d& dir2) {
return (std::abs(dir1.x - dir2.x) < 0.001 &&
std::abs(dir1.y - dir2.y) < 0.001 &&
std::abs(dir1.z - dir2.z) < 0.001);
}
int main() {
std::vector<BossFacesDirData> boss_face_dir_data_vec;
boss_face_dir_data_vec.reserve(boss_faces.size());
std::unordered_map<Vector3d, int> boss_faces_dirs_map;
for (int b = 0; b < boss_faces_dirs.size(); b++) {
boss_faces_dirs_map[boss_faces_dirs[b]] = b + 1;
}
for (int a = 0; a < boss_faces.size(); a++) {
double temp_dir1[3];
getFaceDir(boss_faces[a], temp_dir1);
Vector3d temaa(temp_dir1[0], temp_dir1[1], temp_dir1[2]);
auto it = std::find_if(boss_faces_dirs_map.begin(), boss_faces_dirs_map.end(),
[&](const auto& pair) { return isEqual(pair.first, temaa); });
BossFacesDirData boss_face_dir_data;
boss_face_dir_data.dir_num = (it != boss_faces_dirs_map.end()) ? it->second : 0;
std::copy_n(temp_dir1, 3, boss_face_dir_data.face_dir);
boss_face_dir_data_vec.emplace_back(std::move(boss_face_dir_data));
if (boss_face_dir_data.dir_num > 20) {
break;
}
}
// 输出分组结果
for (const auto& data : boss_face_dir_data_vec) {
std::cout << "Dir Num: " << data.dir_num << ", Face Dir: " << data.face_dir[0] << ", "
<< data.face_dir[1] << ", " << data.face_dir[2] << std::endl;
}
return 0;
}
```
请注意,你需要确保 `Vector3d` 类型已正确定义,并且 `getFaceDir` 函数返回一个正确的方向向量。对于 `isEqual` 函数,你可能需要根据你的具体需求进行适当的修改。
阅读全文