C++ std::map и std::pair<int, int> в качестве ключа
У меня есть следующий код C++:
struct MyStruct
{
int a[2];
int b[2];
};
std::map<std::pair<int, int> , MyStruct*> MyMap;
Теперь я запускаю этот цикл на MyMap:
for(std::map<std::pair<int, int> , MyStruct*>::iterator itr = MyMap.begin(); itr != MyMap.end(); ++itr)
{
std::pair<int, int> p (itr->first().second, itr->first().first);
auto i = MyMap.find(p);
if(i != MyMap.end())
{
//do something
}
}
На самом деле я пытаюсь создать пару, меняя элементы другой пары, например, у меня есть пара ключей (12,16) в MyMap, а также другая пара ключей (16,12); эти два ключа существуют в MyMap, и я точно знаю. Но когда я применяю описанную выше технику, MyMap не возвращает значение, соответствующее замененной клавише, я предполагаю, что MyMap.find(p) соответствует указателю Key; но есть ли способ, чтобы я мог заставить MyMap.find(p) совпадать с соответствующим значением в Key (pair) вместо сопоставления с указателями в Key (pair)? Или я что-то здесь не так делаю?
1 ответ
У вас есть некоторые неточности в вашем коде, скажем, ваш MyStruct
не имеет конструктора копирования, но содержит массивы, itr->first()
в вашем для цикла, в то время как first
нет оператора звонка и др. Следующий код делает то, что вы хотите:
#include <array>
#include <map>
#include <utility>
#include <memory>
#include <stdexcept>
#include <iostream>
struct MyStruct
{
std::array<int, 2> a;
std::array<int, 2> b;
};
template <class T, class U>
std::pair<U, T> get_reversed_pair(const std::pair<T, U>& p)
{
return std::make_pair(p.second, p.first);
}
int main()
{
std::map<std::pair<int, int>, std::shared_ptr<MyStruct>> m
{
{
{12, 16},
std::make_shared<MyStruct>()
},
{
{16, 12},
std::make_shared<MyStruct>()
}
};
std::size_t count = 1;
for(const auto& p: m)
{
try
{
auto f = m.at(get_reversed_pair(p.first));
f -> a.at(0) = count++;
f -> b.at(0) = count++;
}
catch(std::out_of_range& e)
{
}
}
for(const auto& p: m)
{
std::cout << p.first.first << ' ' << p.first.second << " - ";
std::cout << p.second -> a.at(0) << ' ' << p.second -> b.at(0) << std::endl;
}
return 0;
}
Выход:
12 16 - 3 4
16 12 - 1 2