Codeforces Round #480 (Div. 2) C. Posterized

time limit per test
1 second
memory limit per test
256 megabytes
input
standard input
output
standard output

Professor Ibrahim has prepared the final homework for his algorithm’s class. He asked his students to implement the Posterization Image Filter.

Their algorithm will be tested on an array of integers, where the ii-th integer represents the color of the ii-th pixel in the image. The image is in black and white, therefore the color of each pixel will be an integer between 0 and 255 (inclusive).

To implement the filter, students are required to divide the black and white color range [0, 255] into groups of consecutive colors, and select one color in each group to be the group’s key. In order to preserve image details, the size of a group must not be greater than kk, and each color should belong to exactly one group.

Finally, the students will replace the color of each pixel in the array with that color’s assigned group key.

To better understand the effect, here is an image of a basking turtle where the Posterization Filter was applied with increasing kk to the right.

To make the process of checking the final answer easier, Professor Ibrahim wants students to divide the groups and assign the keys in a way that produces the lexicographically smallest possible array.

Input

The first line of input contains two integers nn and kk (1n1051≤n≤1051k2561≤k≤256), the number of pixels in the image, and the maximum size of a group, respectively.

The second line contains nn integers p1,p2,,pnp1,p2,…,pn (0pi2550≤pi≤255), where pipi is the color of the ii-th pixel.

Output

Print nn space-separated integers; the lexicographically smallest possible array that represents the image after applying the Posterization filter.

Examples
input
Copy
4 3
2 14 3 4
output
Copy
0 12 3 3
input
Copy
5 2
0 2 1 255 254
output
Copy
0 1 1 254 254
Note

One possible way to group colors and assign keys for the first sample:

Color 22 belongs to the group [0,2][0,2], with group key 00.

Color 1414 belongs to the group [12,14][12,14], with group key 1212.

Colors 33 and 44 belong to group [3,5][3,5], with group key 33.

Other groups won't affect the result so they are not listed here.

题目说将0-255分成几组最多含有k个数的组,每个组最小的数代表这个组,输出给定组合最小的字典序。。。

其实就是用贪心写,每次看一下右端点往前的点,然后合并,只有小于等于k才能合并,不然只能重新分到一组。。。

简单的贪心,手速问题没写出来。。。

代码:

#include<bits/stdc++.h>

using namespace std;
const int maxn = 1e5+7;
int n, k, p[maxn], vis[300];
void print()
{
    printf("%d", vis[p[0]]);
    for(int i = 1; i < n; i++) printf(" %d", vis[p[i]]);
    printf("\n");
}

void solve()
{
    memset(vis, 255, sizeof(vis));
    for(int i = 0; i < n; i++)
    {
        if(vis[p[i]] == -1)
        {
            int c = max(0, p[i] - k + 1);
            if(vis[c] == -1) {
                for(int j = c; j <= p[i]; j++)
                    vis[j] = c;
            } else {
                int c = p[i];
                while(vis[c] == -1) c--;
                if(p[i] - vis[c] <= (k - 1)) {
                    for(int j = c + 1; j <= p[i]; j++) vis[j] = vis[c];
                } else {
                    for(int j = c + 1; j <= p[i]; j++) vis[j] = c+1;
                }
            }
        }
    }
}

int main()
{
    while(~scanf("%d%d",&n, &k))
    {
        for(int i = 0; i < n; i++) scanf("%d",&p[i]);
        solve();
        print();
    }
    return 0;
}


猜你喜欢

转载自blog.csdn.net/weixin_39792252/article/details/80279922