C++ switch case 匹配字符串

有时候，我们想写出下面这样的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;  
}  

这个函数只有短短的一行，利用递归得到了与上面hash_函数得到的同样值，由于用constexpr声明了函数，因此编译器可以在编译期得出一个字符串的hash值，而这正是关键，既然是编译器就可以得到的整型常量，自然可以放到switch的case标签处了。

于是我们可以写出这样的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  

现在，我们写出的switch语句就好看多了。

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;  
    }  
}

原文：http://blog.csdn.net/yozidream/article/details/22789147