codeforces628 E. Zbazi in Zeydabad 选择枚举对象+用顺序减少重复计算+树状数组优化

问一个n*m的网格中有多少个z

基本想法应该是枚举z，我想的是枚举中间的z，但是好像复杂度太高了，题解的方法是枚举右上角的z，然后预处理出每个点往左最多的和往左下最多的（nm时间）

枚举每个xy，判断以他为右上角的z有多少个，因为对角线可以变短一点，所以要枚举每个对角线上的点，看右边最长能不能到y

至此算法复杂度为o（nmm）

想办法优化，枚举每一个显然不怎么好优化……只有计算对角线上有哪些能到这个y能想办法优化，需要知道这条对角线上有多少点能到y，就是求有多少点加上往右的最大值之后能>y，找到重复计算的部分，同一个对角线上，y+最大值可能很大，但是每次都要算一次，浪费了很多时间，所以要从右边最大的算起，每条对角线上的某个位置，如果以前可以，那么更小的就也可以所以下次就不用再比较，可是这样复杂度依然没有变……需要一个快速求出全部和的方法

就是用树状数组（或者线段树）每次算刚好能到j列的那些，每个对角线保存一个树状数组，就可以利用前面的性质依次计算

好难啊……自己想不出来？

下面是AC代码，初始化的部分有点麻烦了……不过居然AC也是不容易

#include<stdio.h>
#include<string.h>
#include<algorithm>
#include<iostream>
#include<vector>
using namespace std;
int zl[3001][3001];
int zr[3001][3001];
int zd[3001][3001];
int bit[6005][3005];
int n,m;
int gra[3001][3001];
struct node{
	int x;
	int y;
    node(int a,int b):x(a),y(b){};
};
vector<node> value[3005];//加了之后为j的 

int sum(int k,int i){
	int s = 0;
	while(i > 0){
		s += bit[k][i];
		i -= i & -i;
	}
	return s;
}
    
void add(int j,int i,int x){
	while(i <= m){
		bit[j][i] += x;
		i += i & -i;
	}
}


void init()
{
	for(int i = 1; i <= n; i++){//直接加后面一个就行了，我这样写烦了 
		int tem = 0;
	    for(int j = m; j >= 1; j--){
	    	if (gra[i][j] == 'z'){
	    		tem++;
	    		zr[i][j] = tem;
			}
			else tem = 0;
		}
    }
    for(int i = 1; i <= n; i++){
    	int tem = 0;
    	for(int j = 1; j <= m; j++){
    		if (gra[i][j] == 'z'){
    			tem++;
    			zl[i][j] = tem;
			}
			else tem = 0;
		}
	}
	for(int i = 1; i <= n; i++){
		int tem = 0;
		for(int k = 0; i - k >= 1 && 1 + k <= m; k++){
			if (gra[i - k][1 + k] == 'z'){
				tem++;
				zd[i - k][1 + k] = tem;
			} 
			else tem = 0;
		}
	}
	for(int j = 2; j <= m; j++){
		int tem = 0;
		for(int k = 0; (n - k) >= 1 && j + k <= m; k++){
			if (gra[n - k][j + k] == 'z'){
				tem++;
				zd[n - k][j + k] = tem;
			}
			else tem = 0;
		}
	}
	for(int i = 1; i <= n; i++)
	for(int j = 1; j <= m; j++)
	  value[j + zr[i][j] - 1].push_back(node(i,j));
}

int main()
{
	long long ans = 0;
	memset(bit ,0,sizeof(bit));//清空 
	scanf("%d %d",&n,&m);
	getchar();
	//输入 
	for(int i = 1; i <= n; i++){
		for(int j = 1; j <= m; j++)
		    scanf("%c",&gra[i][j]);
		getchar(); 
	}
	init();
	for(int j = m; j >= 1; j--){
	    for(int i = 0; i < value[j].size();i++)
	        add(value[j][i].x + value[j][i].y,value[j][i].y,1);// 
	for(int i = 1; i <= n; i++){
	   int len = min(zl[i][j],zd[i][j]); 
	ans += (sum(i + j,j) - sum(i + j,j - len));
    }
}
	printf("%I64d\n",ans);
	return 0;
	}