SPOJ_1329 KPMATRIX

http://www.spoj.pl/problems/KPMATRIX/

题意：

给你一个N*M的矩阵，N,M<=250，所有和 >=A && <=B的子矩阵的个数。

思路：

首先我们枚举子矩阵的上下边，复杂度为O(n ^ 2)， 然后我们处理出前缀和sum[x] , 当我们在找以x为子矩阵的右边界的子矩阵的时候，就是要找有多少个i，使得 A<= sum[x] - sum[i] <=B ，也就是 sum[x] - B <= sum[i] <= sum[x] - A ，也就是说在某个位置x，我们只要找到它前面一个有多少个i，满足sum[x] - B <= sum[i] <= sum[x] - A ，这个查找是可以在O( logm )的时间内找到的（用一个数状数组维护即可），但是因为 -10^9 <= sum[i] <=10^9 ，所以我们还需要对所有的数进行离散化处理。这样本题的总时间复杂度就是:O(n^2*(m + m*logm) ) 。

代码：

#include <stdio.h>
#include <string.h>
#include <algorithm>

const int NN = 260 ;
int N,M, A, B;
int map[NN][NN] ;
int sum[NN] ;
int sumB[NN] , sumA[NN] ;
int xx[NN*3] ;
int C[3 * NN] ;


int find( int val , int l ,int r ){
    while( l<=r ){
        int mid = (l + r) >> 1  ;
        if( xx[mid] == val )    return mid ;
        if( xx[mid] < val ) l = mid + 1 ;
        else            r = mid - 1 ;
    }
    return -1 ;
}

inline int lowbit(int a){ return a&(-a)  ; }

void add(int a , int n){
        while( a<=n ){
            C[a] ++ ; a += lowbit(a) ;
        }
}

int cal_sum(int n){
    int res = 0 ;
    while( n>=1 ){
        res += C[n]  ;
        n -= lowbit(n) ;
    }
    return res ;
}
void solve(){
    int ans = 0 ;
    for(int i=1;i<=N;i++){
        for(int j=i;j<=N;j++){
            int cnt = 1 ;
            sum[0] = 0  ;
            sumA[0] = sum[0] - A ;
            sumB[0] = sum[0] - B - 1 ;
            xx[cnt++ ] = sum[0]  ;
            xx[cnt++ ] = sumA[0] ;
            xx[cnt++ ] = sumB[0] ;

            for(int k=1;k<=M;k++){
                sum[k] = sum[k-1] + map[j][k] - map[i-1][k] ;
                sumA[k] = sum[k] - A ;
                sumB[k] = sum[k] - B - 1;
                xx[cnt ++] = sum[k]  ;
                xx[cnt ++] = sumB[k] ;
                xx[cnt ++] = sumA[k] ;
            }
            std::sort( xx + 1 , xx + cnt ) ;
            int tot = 2 ;
            for(int i=2;i<cnt;i++){
                if( xx[i] == xx[i-1] )  continue ;
                xx[ tot++ ] = xx[i] ;
            }
            tot -- ;
            memset( C , 0 ,sizeof(C) );
            add( find( sum[0] , 1 , tot ) , tot ) ;
            for(int i=1;i<=M;i++){
                int e = find( sumA[i] , 1 , tot ) ;
                int s = find( sumB[i] , 1 , tot ) ;
                ans += cal_sum(e) - cal_sum(s) ;
                add( find( sum[i] ,1 , tot) , tot ) ;
            }
        }
    }
    printf("%d\n",ans);
}

int main(){
    while( scanf("%d %d",&N,&M) == 2 ){
        for(int j=1;j<=M;j++)   map[0][j] = 0 ;
        for(int i=1;i<=N;i++){
            for(int j=1;j<=M;j++){
                scanf("%d",&map[i][j]) ;
                map[i][j] = map[i-1][j] + map[i][j] ;
            }
        }
        scanf("%d %d",&A,&B);
        solve() ;
    }
    return 0 ;
}