BZOJ-1452: [JSOI2009]Count(二维BIT)

题目:http://www.lydsy.com/JudgeOnline/problem.php?id=1452

二维BIT直接暴力维护即可,不用多说什么饿。

代码:

#include <cstdio>

#include <algorithm>

#include <cstring>

 

using namespace std ;

 

#define lowbit( x ) ( ( - x ) & x )

#define MAXN 310

#define MAXC 110

 

struct BIT {

     

    int a[ MAXN ][ MAXN ] , N , M ;

     

    void Init( int _N , int _M ) {

        memset( a , 0 , sizeof( a ) ) ;

        N = _N , M = _M ;

    }

     

    void Add( int x , int y , int value ) {

        for ( ; x <= N ; x += lowbit( x ) ) {

            for ( int i = y ; i <= M ; i += lowbit( i ) ) {

                a[ x ][ i ] += value ;

            }

        }

    }

     

    int Sum( int x , int y ) {

        int ret = 0 ;

        for ( ; x ; x -= lowbit( x ) ) {

            for ( int i = y ; i ; i -= lowbit( i ) ) {

                ret += a[ x ][ i ] ;

            }

        }

        return ret ;

    }

     

} bit[ MAXC ] ;

 

int n , m , q , c[ MAXN ][ MAXN ] , maxc = 0 ;

 

int main(  ) {

    scanf( "%d%d" , &n , &m ) ;

    for ( int i = 0 ; i ++ < n ; ) {

        for ( int j = 0 ; j ++ < m ; ) {

            scanf( "%d" , &c[ i ][ j ] ) ;

            maxc = max( maxc , c[ i ][ j ] ) ;

        }

    }

    for ( int i = 0 ; i ++ < maxc ; ) bit[ i ].Init( n , m ) ;

    for ( int i = 0 ; i ++ < n ; ) {

        for ( int j = 0 ; j ++ < m ; ) {

            bit[ c[ i ][ j ] ].Add( i , j , 1 ) ;

        }

    }

    scanf( "%d" , &q ) ;

    while ( q -- ) {

        int p ; scanf( "%d" , &p ) ;

        if ( p == 1 ) {

            int x , y , col ; scanf( "%d%d%d" , &x , &y , &col ) ;

            bit[ c[ x ][ y ] ].Add( x , y , - 1 ) ;

            bit[ c[ x ][ y ] = col ].Add( x , y , 1 ) ;

        } else {

            int x0 , x1 , y0 , y1 , col ;

            scanf( "%d%d%d%d%d" , &x0 , &x1 , &y0 , &y1 , &col ) ;

            int ans = bit[ col ].Sum( x1 , y1 ) ;

            ans -= ( bit[ col ].Sum( x0 - 1 , y1 ) + bit[ col ].Sum( x1 , y0 - 1 ) ) ;

            ans += bit[ col ].Sum( x0 - 1 , y0 - 1 ) ;

            printf( "%d\n" , ans ) ;

        }

    }

    return 0 ;

}
    原文作者:AmadeusChan
    原文地址: https://www.jianshu.com/p/f9a0a59e4b3e
    本文转自网络文章,转载此文章仅为分享知识,如有侵权,请联系博主进行删除。
点赞