PAT 乙级 1066.图像过滤 C++/Java

题目来源

图像过滤是把图像中不重要的像素都染成背景色，使得重要部分被凸显出来。现给定一幅黑白图像，要求你将灰度值位于某指定区间内的所有像素颜色都用一种指定的颜色替换。

输入格式：

输入在第一行给出一幅图像的分辨率，即两个正整数 M 和 N（0），另外是待过滤的灰度值区间端点 A 和 B（0）、以及指定的替换灰度值。随后 M 行，每行给出 N 个像素点的灰度值，其间以空格分隔。所有灰度值都在 [0, 255] 区间内。

输出格式：

输出按要求过滤后的图像。即输出 M 行，每行 N 个像素灰度值，每个灰度值占 3 位（例如黑色要显示为 000），其间以一个空格分隔。行首尾不得有多余空格。

输入样例：

3 5 100 150 0
3 189 254 101 119
150 233 151 99 100
88 123 149 0 255

 

输出样例：

003 189 254 000 000
000 233 151 099 000
088 000 000 000 255

C++实现：

#include <iostream>
#include <iomanip>
#include <vector>
using namespace std;
//1066：图像过滤
int main() {
    int m, n, a, b, replace;
    cin >> m >> n >> a >> b >> replace;
    vector<vector<int> > pic(m, vector<int>(n));
    for (int i = 0; i < m; i++) {
        for (int j = 0; j < n; j++) {
            cin >> pic[i][j];
            if (pic[i][j] >= a && pic[i][j] <= b) {
                pic[i][j] = replace;
            }
        }
    }
    for (int i = 0; i < m; i++) {
        for (int j = 0; j < n; j++) {
            if (j != 0) {
                cout << ' ';
            }
            cout << setw(3) << setfill('0') << pic[i][j];
        }
        cout << endl;
    }
    return 0;
}

 

 

java实现：

最后一个测试点超时

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;

public class Main {
    public static void main(String[] args) throws IOException {
        BufferedReader in = new BufferedReader(new InputStreamReader(System.in));
        String[] s = in.readLine().split("\\s+");
        int[] m = new int[5];
        for(int i = 0; i < 5; i++){
            m[i] = Integer.parseInt(s[i]);
        }
        for (int i = 0; i < m[0]; i++) {
            String[] str = in.readLine().split("\\s+");
            for (int j = 0; j < m[1]; j++) {
                int p = Integer.parseInt(str[j]);
                if (p >= m[2] && p <= m[3]) {
                    p = m[4];
                }
                if (j != 0)
                    System.out.print(" ");
                System.out.printf("%03d", p);
            }
            System.out.println();
        }
    }
}