661. Image Smoother的C++解法

版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/musechipin/article/details/81737637

题目本身很简单,这里主要考虑如何尽量占用小的空间。原图给出的数值范围是(0,255),也就是只占低八位,因此考虑把平滑结果保存到高八位上,这样就不需要再开辟新的存储空间了。

class Solution {
public:
    vector<vector<int>> imageSmoother(vector<vector<int>>& M) {
        for (int i = 0; i < M.size(); ++i) {
            for (int j = 0; j < M[0].size(); ++j) {
                int sum = 0;
                int count = 0;
                for (int k = max(0, i-1); k < min(i+2, static_cast<int>(M.size())); ++k) {
                    for (int l = max(0, j-1); l < min(j+2, static_cast<int>(M[0].size())); ++l) {
                        sum += M[k][l] & 0xFF;
                        count += 1;
                    }
                }
                M[i][j] |= (sum / count) << 8;
            }
        }
        for (int i = 0; i < M.size(); ++i) {
            for (int j = 0; j < M[0].size(); ++j) {
                M[i][j] >>= 8;
            }
        }
        return M;
    }
};

1. & 0xFF是取低八位操作。

2. >>右移八位。

猜你喜欢

转载自blog.csdn.net/musechipin/article/details/81737637