其他分享
首页 > 其他分享> > leetcode 1901. 找出顶峰元素 II

leetcode 1901. 找出顶峰元素 II

作者:互联网

1901. 找出顶峰元素 II

解题思路

这里有点投机取巧了,用了标准库中的max_element,先找到每一行的最大值,然后将索引i, j, max_val保存到数列中,然后对数据项取max_val,找到元素位置,接着返回。

代码

class Solution {
public:
    vector<int> findPeakGrid(vector<vector<int>>& mat) {
        vector<vector<int>> positions;
        for (int i = 0; i < mat.size(); i++) {
            auto item = max_element(mat[i].begin(), mat[i].end());
            positions.push_back({i, static_cast<int>(item - mat[i].begin()), *item});
        }
        auto item = *max_element(positions.begin(), positions.end(), CompareVal);
        return {item[0], item[1]};
    }
private:
    static bool CompareVal(vector<int>& item1, vector<int>& item2) {
        return item1[2] < item2[2];
    }
};

标签:begin,mat,max,II,item,vector,positions,leetcode,1901
来源: https://blog.csdn.net/u013395337/article/details/121091313