您的位置:首页 > 其它

STL map中key为结构体的用法

2014-06-06 18:23 267 查看
原文链接:http://www.cppblog.com/Apan/articles/75267.html

最近在使用stl中的map容器时,碰到key为结构体的情况,总结如下,以便提醒自己。

我的使用情景是,我需要根据不同的比例尺、道路类型这两个参数获取到对应的道路宽度,由于我是使用map解决这个问题的,自然而然的就以比例尺、道路类型这两个参数为key,道路宽度为value,建立的key如下:

typedef struct tagRoadKey
{
int nType;
int nScale;
}ROADKEY;
但是编译的时候,报了这个错误

d:\program files\microsoft visual studio\vc98\include\functional(86) : error C2784: 'bool __cdecl std::operator <(const class std::multimap<_K,_Ty,_Pr,_A> &,const class std::multimap<_K,_Ty,_Pr,_A> &)' : could not deduce template
argument for 'const

class std::multimap<_K,_Ty,_Pr,_A> &' from 'const struct tagRoadKey'

说实话,当初不太明白这是什么错误,但从个人经验来判断,问题肯定出在这个key上面,后来google下,看到了别人写的文章,才知道原因,原来map中的key默认是以less<>升序对元素排序(排序准则也可以修改),也就是说key必须具备operator<对元素排序,而平常我们的用的基本上都是基本类型元素作为key,所以就不存在这个问题了,更详细的解释请看C++标准程序库一书,第六章,set容器章节。

改正后的结构体如下:

typedef struct tagRoadKey
{
int nType;
int nScale;

bool operator <(const tagRoadKey& other) const
{
if (nType < other.nType)        //类型按升序排序
{
return true;
}
else if (nType == other.nType)  //如果类型相同,按比例尺升序排序
{
return nScale < other.nScale;
}

return false;
}
}ROADKEY;


完整代码如下:

//////////.h///////
#ifndef _CROADWIDTHMNG_H_
#define _CROADWIDTHMNG_H_
#include <map>

using namespace std;

/*
说明:根据当前比例尺、道路类型获取对应的道路宽度
*/
typedef struct tagRoadKey
{
int nType;
int nScale;

bool operator <(const tagRoadKey& other) const
{
if (nType < other.nType)        //类型按升序排序
{
return true;
}
else if (nType == other.nType)  //如果类型相同,按比例尺升序排序
{
return nScale < other.nScale;
}

return false;
}
}ROADKEY;

struct tagRoadInfo
{
tagRoadKey roadKey;
int        nRoadWidth;
};

class CRoadWidthMng
{
public:
CRoadWidthMng();
virtual ~CRoadWidthMng();

public:
int GetRoadWidth(int nRoadType, int nScale); //根据道路类型、比例尺获取宽度

private:
void Init();

private:
const int  MIN_SCALE;           //最小的比例尺
const int  DEFAULT_ROAD_WIDTH;  //如没有找到,返回默认值

map<ROADKEY, int>m_roadMap;
};

#endif

////////.cpp///////
#include "CRoadWidthMng.h"

tagRoadInfo roadInfoItem[] =
{
///////////高速公路//////////////
{
{
10,
12
},
16
},
{
{
10,
11
},
12
},

{
{
10,
10
},
6
},
{
{
10,
9
},
3
},
///////国道/////////////////
{
{
12,
12
},
12
},
{
{
12,
11
},
8
},

{
{
12,
10
},
6
},
{
{
12,
9
},
4
},
///////省道/////////////////
{
{
14,
12
},
10
},
{
{
14,
11
},
10
},

{
{
14,
10
},
6
},
{
{
14,
9
},
4
},
///////铁路/////////////////
{
{
21,
12
},
1
},
{
{
21,
11
},
1
},

{
{
21,
10
},
1
},
{
{
21,
9
},
1
},
};

CRoadWidthMng::CRoadWidthMng()
:MIN_SCALE(6), DEFAULT_ROAD_WIDTH(5)
{
Init();
}

CRoadWidthMng::~CRoadWidthMng()
{
m_roadMap.clear();
}

void CRoadWidthMng:: Init()
{
int nNum = sizeof(roadInfoItem) / sizeof(roadInfoItem[0]);

for (int i = 0; i < nNum; ++i)
{
m_roadMap.insert(make_pair(roadInfoItem[i].roadKey, roadInfoItem[i].nRoadWidth));
}
}

int CRoadWidthMng:: GetRoadWidth(int nRoadType, int nScale)
{
if (nScale < MIN_SCALE)
{
nScale = MIN_SCALE;
}

map<ROADKEY, int>::iterator itor;

int nValue;
ROADKEY roadkey;
roadkey.nType = nRoadType;
roadkey.nScale = nScale;

itor = m_roadMap.find(roadkey);

if (itor != m_roadMap.end())
{
nValue =  itor->second/*nRoadWidth*/;
}
else
{
nValue =  DEFAULT_ROAD_WIDTH;
}

return nValue;
}
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: