washiの備忘録: std::mapの使い方

Collada読み込みにおいて、stlのmapを使ったほうがよさそうな場面に出くわしたので、使い方をチェック。キーはstd::stringが手っ取り早そう。要素の挿入・参照方法は、イテレータを使用しないシンプルなものを採用した。

01#include <iostream>
02#include <string>
03#include <map>
04using namespace std;
05 
06int main(int argc, char **argv)
07{
08   map<string, int> tbl;
09 
10   // 挿入
11   tbl["hoge"] = 0;
12   tbl["piyo"] = 10;
13   tbl["fuga"] = 20;
14 
15   // 参照
16   cout << tbl["hoge"] << endl;
17   cout << tbl["piyo"] << endl;
18   cout << tbl["fuga"] << endl;
19   return 0;
20}

ふと気になってoperator[]に無効なキーを与えた場合はどうなるのか実験してみた。予想だと例外を吐いて、戻り値なしかなーとか思ってたのですが…

01#include <iostream>
02#include <string>
03#include <map>
04using namespace std;
05 
06int main(int argc, char **argv)
07{
08   map<string, int> tbl;
09 
10   // 挿入
11   tbl["hoge"] = 0;
12   tbl["piyo"] = 10;
13   tbl["fuga"] = 20;
14 
15   // 参照
16   cout << tbl["hoge"] << endl;//0
17   cout << tbl["piyo"] << endl;//10
18   cout << tbl["fuga"] << endl;//20
19 
20   // 無効なキーにアクセスした場合
21   int ret = tbl["invalid key"];
22   cout << ret << endl;//0 <- intの初期値0が新しいキー"invalid key"と一緒に作成される模様
23   return 0;
24}

結果から言うと、普通にアクセスできて、0が返ってきた。どうやら、無効なキーを使ってoperator[]経由でマップにアクセスした場合は、そのキーを元に新しい要素（ここではint）が作成される模様。この場合intの初期値である0で値が初期化され、("invalid key", 0)という要素がマップに追加されたというわけだ。

私がやりたいことは、例えばstd::stringの"VERTEX"というキーがあるかどうか調べて、あればそのキーに関連した値（unsigned intで表現されるインデックス配列のオフセット値）を取り出し、なければ無視するということなので、このままでは実現不可能ですね。次はfindを使うやり方を模索しよう。

findを使用した検索

01#include <iostream>
02#include <string>
03#include <map>
04using namespace std;
05typedef map<string, int> Table;
06 
07int main(int argc, char **argv)
08{
09   Table tbl;
10 
11   // 挿入
12   tbl["hoge"] = 123;
13   tbl["piyo"] = 456;
14   tbl["fuga"] = 789;
15 
16   // 検索
17   Table::iterator ret = tbl.find("hoge");
18   if(ret != tbl.end())
19   {
20      // 見つかった
21      cout << "Key: " << ret->first;
22      cout << ", Value: " << ret->second << endl;
23   }
24   else
25   {
26      // 見つからなかった
27      cout << "Not Found." << endl;
28   }
29 
30   return 0;
31}

イテレータ宣言が長くなるのでmap<string, int>をTableという名前に定義しなおしています。個人的にはこういうやり方が好き。"hoge"に対応する値が見つかった場合はそいつを指すイテレータを、見つからなかった場合は末端tbl.end()を指すイテレータを返します。

ここでは、int値だけが欲しいのでit->secondで目的の値を取り出せます。it->firstだとキー値を取り出せます。

あとはこいつを関数化してあげればいいだけ。

01/**
02 * std::map<std::string, int>からキーに対応する値を取り出す。
03 * @param out 取り出した値の格納先。見つからなかった場合は何も変更しない。
04 * @param keyword キー値
05 * @param tbl 検索するマップ
06 * @return キーに対応する値がある場合は真。
07 */
08bool FindValueFromMap(int *out,
09                      const string &keyword,
10                      const Table &tbl)
11{
12   Table::const_iterator it = tbl.find(keyword);
13 
14   if(it != tbl.end())
15   {
16      *out = it->second;
17      return true;
18   }
19   return false;
20}

始めは戻り値をそのままintにしよう思っていたのですが、よく考えたら値が見つからなかった場合はどないすんねんとなったので、boolに変更。ポインタ経由で値をやり取りするようにしました。例外とかに変更したほうがいいのかなぁ。

んで、この関数はtemplateで汎化できるんじゃね？　と思って以下のようなコードを書いたが、コンパイルが通らなかった。

01template<typename KeyType, typename MappedType>
02bool FindValueFromMapT(MappedType *out,
03                       const KeyType &keyword,
04                       const std::map<KeyType, MappedType> &table)
05{
06   std::map<KeyType, MappedType>::const_iterator it = table.find(keyword);
07   if(it != table.end())
08   {
09      *out = it->second;
10      return true;
11   }
12   return false;
13}

エラーメッセージ

1main.cpp: In function `bool FindValueFromMapT(MappedType*, const KeyType&, const std::map<KeyType, MappedType, std::less<_Key>, std::allocator<std::pair<const _Key, _Tp> > >&)':
2main.cpp:60: error: expected `;' before "it"
3main.cpp:61: error: `it' undeclared (first use this function)
4main.cpp:61: error: (Each undeclared identifier is reported only once for each function it appears in.)

どうもイテレータ宣言方法で失敗しているっぽいが、よくわからん。まあここまでやる必要もなさそうなので、テンプレート化はあきらめた。

さらに追記。上記のテンプレートだがVC++2008でコンパイルしたところ何の問題もなく通った。呼び出しも予想通りの結果を返してくれた･･･。g++のコンパイルオプションあたりが怪しそう！　ちょっと調べてみよう。

C++テンプレートのFAQという記事を発見。g++が厳しいというよりは、VC++が緩いというべきか。厳密なテンプレートの記述を行えばコンパイルは通りそう。

コンパイル通りました！

01template<typename KeyType, typename MappedType>
02bool FindValueFromMapT(MappedType *out,
03                       const KeyType &key,
04                       const std::map<KeyType, MappedType> &table)
05{
06   typename std::map<KeyType, MappedType>::const_iterator it = table.find(key);
07   if(it != table.end())
08   {
09      *out = it->second;
10      return true;
11   }
12   return false;
13}

前回との違いは

typename std::map<KeyType, MappedType>::const_iterator it = table.find(key);

の部分、これだけ！なぜこれをつけるとうまくいくかについては、現在確認中ですが、dependant nameというのが関係してそう。英語なのでじっくり読んでいきます。

washiの備忘録

2010年6月7日月曜日

std::mapの使い方

2 件のコメント: