使用键作为结构添加地图

时间:2017-01-11 03:09:27

标签: c++ dictionary stl

    typedef struct A{
      string id;
      long date;
      // operator = and < overloading
    }Test;

Map<Test, double> testMap[2];

在代码中,testMap数组中包含带有一些业务逻辑的键和值。

我需要为每个A.id计算两个地图的总重复值。请注意,总和应仅基于A.id,而不是整个结构键。
为了达到这个目的,我可以使用for循环来应用正常的蛮力方法并得到结果。

但我试图找到可以优化代码的此问题的替代解决方案。请建议。

1 个答案:

答案 0 :(得分:1)

执行此操作的一种方法是应用嵌套std::accumulate两次,一次总结数组,并为每个数组总结 map 内容:

struct Test
{
    string id;
    long date;
    bool operator<(Test const& test) const
    {
        if(date == test.date)
            return id < test.id;
        return date < test.date;
    }
};

double sum_per_id(std::array<std::map<Test, double>, 2> const& testMapArray,
    std::string const& id)
{
    return std::accumulate(std::begin(testMapArray), std::end(testMapArray), 0.0,
    [&id](double d, std::map<Test, double> const& testMap)
    {
        return d + std::accumulate(std::begin(testMap), std::end(testMap), 0.0,
        [&id](double d, std::map<Test, double>::value_type const& p)
        {
            if(id == p.first.id)
                return d + p.second;
            return d;
        });
    });
}

int main()
{
    std::array<std::map<Test, double>, 2> testMapArray;

    testMapArray[0][{"A", 0}] = 0.1;
    testMapArray[0][{"B", 1}] = 0.2;

    testMapArray[1][{"A", 2}] = 0.3;
    testMapArray[1][{"B", 3}] = 0.4;

    std::cout << "sum: " << sum_per_id(testMapArray, "A") << '\n';
}

<强>输出:

sum: 0.4