有时候,我们想写出下面这样的switch语句:
const char* str = "first"; switch(str){ case "first": cout << "1st one" << endl; break; case "second": cout << "2nd one" << endl; break; case "third": cout << "3rd one" << endl; break; default: cout << "Default..." << endl; }
但是在c++中,是不能用字符串来作为case的标签的;于是,很疑惑,我们只能用其他的办法来实现这样的需求。
但幸运的是,c++11引入了constexpr和自定义文字常量,将这两个新特性结合,我们实现出看上去像上面这样的代码。
基本思路为:
1、定义一个hash函数,计算出字符串的hash值,将字符串转换为1个整数;
typedef std::uint64_t hash_t; constexpr hash_t prime = 0x100000001B3ull; constexpr hash_t basis = 0xCBF29CE484222325ull; hash_t hash_(char const* str) { hash_t ret{basis}; while(*str){ ret ^= *str; ret *= prime; str++; } return ret; }
2、利用c++11自定义文字常量的语法,定义一个constexpr函数,switch的case标签处调用这个constexpr函数。如下
constexpr hash_t hash_compile_time(char const* str, hash_t last_value = basis) { return *str ? hash_compile_time(str+1, (*str ^ last_value) * prime) : last_value; }
于是我们可以写出这样的swich语句:
void simple_switch2(char const* str) { using namespace std; switch(hash_(str)){ case hash_compile_time("first"): cout << "1st one" << endl; break; case hash_compile_time("second"): cout << "2nd one" << endl; break; case hash_compile_time("third"): cout << "3rd one" << endl; break; default: cout << "Default..." << endl; } }
这个实现中,hash_compile_time("first")是编译器计算出来的一个常量,因此可以用作case标签;而且如果出现了hash值冲突,编译器回给出错误提示。
3、上面的语法还不够漂亮,利用自定义文字常量,重载一个operator如下:
constexpr unsigned long long operator "" _hash(char const* p, size_t) { return hash_compile_time(p); }
现在我们可以写这样的文字常量,用“_hash”作为自定义文字常量的后缀,编译器看到这个后缀结尾的文字常量,就会去调用我们重载的operator,得到和调用hash_compile_time是一样的值,但看起来舒服多了:
"first"_hash
void simple_switch(char const* str) { using namespace std; switch(hash_(str)){ case "first"_hash: cout << "1st one" << endl; break; case "second"_hash: cout << "2nd one" << endl; break; case "third"_hash: cout << "3rd one" << endl; break; default: cout << "Default..." << endl; } }
全文完。希望我讲清楚了。