提问者:小点点

起始值非零的枚举类上的运算符


我需要高效的运算符枚举类,问题是枚举不从零开始计算困难。

这是有问题的枚举:

#include <cstdint>
#include <iostream>

enum class CardRank : std::uint16_t
{
    Two = 2,
    Three,
    Four,
    Five,
    Six,
    Seven,
    Eight,
    Nine,
    Ten,
    Jack,
    Queen,
    King,
    Ace     // = 14
};

我希望此枚举的运算符接受任何整数值(直到的限制)。

经过一段时间的挣扎,我想出了以下算法:

inline CardRank operator+(const CardRank& rank, const std::uint16_t num)
{
    std::uint16_t need = static_cast<std::uint16_t>(rank) + num;

    if (need < 15)
        return static_cast<CardRank>(need);

    std::uint16_t result = need % 14 + need / 14;

    return static_cast<CardRank>(result > 14 ? result - 13 : result);
}

这是测试代码:

int main()
{
    CardRank rank = CardRank::King;

    // = CardRank::Ace
    std::cout << static_cast<std::uint16_t>(rank + 1) << std::endl;

    // = CardRank::Four
    std::cout << static_cast<std::uint16_t>(rank + 17) << std::endl;

    // = CardRank::Two
    std::cout << static_cast<std::uint16_t>(rank + 41) << std::endl;

    // = CardRank::Ten
    std::cout << static_cast<std::uint16_t>(rank + 114) << std::endl;

    std::cin.get();
}

计算是正确的,运算符工作得很好,但问题是我只想加2个数字,但事实证明,计算量需要得到的结果是不值得运算符。

有没有一些传统的更有效的方法来使这变得更简单?

编辑:顺便说一句,我不能将王牌添加到开头并将其设置为1,因为排序算法从最高到最小的卡片排序。这意味着王牌必须在国王之上。

编辑

多亏了答案,我想出了以下一句话:

inline CardRank operator+(const CardRank& rank, const std::uint16_t num)
{
    constexpr std::uint16_t offset = 2;
    constexpr std::uint16_t cards = 13;

    return static_cast<CardRank>((static_cast<std::uint16_t>(rank) - offset + num) % cards + offset);
}

共1个答案

匿名用户

问题是枚举不是从零开始的,这使得计算变得困难。

这是真的——所以首先要做的是减去2,使它从0开始。记住在最后加上2。这使得事情变得简单。

inline CardRank operator+(const CardRank& rank, const std::uint16_t num)
{
    std::uint16_t need = (static_cast<std::uint16_t>(rank) - 2 + num) % 13 + 2;

    return static_cast<CardRank>(need);
}