diff --git a/a8/commonrank.h b/a8/commonrank.h index 1810208..085bd41 100644 --- a/a8/commonrank.h +++ b/a8/commonrank.h @@ -81,31 +81,31 @@ namespace a8 }; template - a8::CommonRank& ForceCreateCommonRankList(std::map>& rank_list, + a8::CommonRank* ForceCreateCommonRankList(std::map>& rank_list, long long key, std::function cmp_func, size_t max_num) { auto itr = rank_list.find(key); if (itr != rank_list.end()) { - return itr->second; + return &itr->second; } rank_list[key] = a8::CommonRank(cmp_func, max_num); - return rank_list[key]; + return &rank_list[key]; } template - static a8::CommonRank& ForceCreateCommonRankList(std::map>& rank_list, + static a8::CommonRank* ForceCreateCommonRankList(std::map>& rank_list, const std::string& key, std::function cmp_func, size_t max_num) { auto itr = rank_list.find(key); if (itr != rank_list.end()) { - return itr->second; + return &itr->second; } rank_list[key] = a8::CommonRank(cmp_func, max_num); - return rank_list[key]; + return &rank_list[key]; } }