hdu 3648 Median Filter

原創

2020-02-22 21:05

我们按“S”型枚举每一个点，如左图：即右->下->左->下->右....

用一个集合存储当前子正方形内的数，下一次查找时，插入新的一列，删除旧的列。如右图：当前枚举点150(三角标记)，黑色正方形的值保存一个集合中，求出中位数。当枚举的点有150移动到125时，我们把蓝色椭圆内的点插入，把黄色椭圆内的点删掉，此时集合中的元素即为红色正方形内的元素，可求出新的中位数。不断如此操作，直到结束。

首先分析一下复杂度：子正方形边长L=2*r+1，N=L*L枚举每个点，需要(500-L)* (500-L)

下面主要的问题是如何快速求中位数？

如果每次排序找中位数，需要Nlog(N)，复杂度为：(500-L)* (500-L)* Nlog(N)，9<=N<250000，这么大的复杂度会超时。所以我们在查中位数时尽量降低复杂度。

树状数组和线段树都可以实现查找第k大数。线段树插入、查找均为log(N)，树状数组取决于二进制中1的个数，所以实际比log(N)还要快。

若用线段树，每次O(L*lgN)的插入，O(L*lgN)的删除，O(lgN)的查找。但写完后超时，好像线段树系数很大，实现的时候不能完全达到log

用树状数组，每次O(L *lgN)的插入，O(L*lgN)的删除，O((lgN)^2)的查找，C++1.5秒可过。在查找时如果不用二分查找，也可优化为O(lgN)的查找，效率会更高。

这道题输出很坑爹，最后有空格。。。。

#include <iostream>
#include <cstdio>
#include <cstring>
using namespace std;
int rr, c[1111111];
int lowbit(int x)
{
    return x & (-x);
}
void add(int i, int val)
{
    while(i <= rr){
        c[i] += val;
        i += lowbit(i);
    }
}
int sum(int i)
{
    int s = 0;
    while(i > 0){
        s += c[i];
        i -= lowbit(i);
    }
    return s;
}
int mat[555][555];
int ans[555][555];
int mm;
int Bin()
{
    int l = 1, r = rr;
    while(l < r){
        int m = (l + r) >> 1;
        if(sum(m) >= mm) r = m;
        else l = m + 1;
    }
    return l - 1;
}

int main()
{
    //freopen("input.txt", "r", stdin);
    //freopen("output.txt", "w", stdout);
    int n, r;
    while(scanf("%d %d", &n, &r) == 2){
        if(!n && !r) {
            break;
        }
        r = (r << 1) + 1;
        mm = (r * r + 1) >> 1;
        rr = -1;
        for(int i = 0; i < n; i++)
            for(int j = 0; j < n; j++){
                scanf("%d", &mat[i][j]);
                rr = max(rr, mat[i][j] + 1);
            }
        memset(c, 0, sizeof(c));
        for(int i = 0; i < r; i++){
            for(int j = 0; j < r - 1; j++){
                add(mat[i][j] + 1, 1);
            }
        }
        for(int i = 0; ; ){
            for(int j = r - 1; j < n; j++){
                for(int k = 0; k < r; k++)
                    add(mat[i + k][j] + 1, 1);
                ans[i][j - (r - 1)] = Bin();
                for(int k = 0; k < r; k++)
                    add(mat[i + k][j - (r - 1)] + 1, -1);
            }
            for(int j = n - 1; j >= n - (r - 1); j--)
                add(mat[i][j] + 1, -1);
            i++;
            if(i + (r - 1) >= n) break;
            for(int j = n - 1; j >= n - (r - 1); j--)
                add(mat[i + (r - 1)][j] + 1, 1);

            for(int j = n - r; j >= 0; j--){
                for(int k = 0; k < r; k++)
                    add(mat[i + k][j] + 1, 1);
                ans[i][j] = Bin();
                for(int k = 0; k < r; k++)
                    add(mat[i + k][j + (r - 1)] + 1, -1);
            }
            for(int j = 0; j < (r - 1); j++)
                add(mat[i][j] + 1, -1);
            i++;
            if(i + (r - 1) >= n) break;
            for(int j = 0; j < (r - 1); j++)
                add(mat[i + (r - 1)][j] + 1, 1);
        }
        for(int i = 0; i <= n - r; i++){
            for(int j = 0; j <= n - r; j++){
                //if(j) printf(" ");
                printf("%d ", ans[i][j]);
            }
            printf("\n");
        }
    }
    return 0;
}