Problem Description
在打败了易基•普罗布朗、诺姆•普罗布朗之后,小茗同学开始挑战哈德•普罗布朗。
一番交战之后,哈德展开了一大波攻击。小茗同学为了抵御攻击,一边放魔法阵一边放魔法阵,然后他也不知道自己一共放了几个魔法阵。回收魔法阵是需要花费时间的,为了抵御下一波攻击,小茗同学需要知道自己共放了几个魔法阵,由于情况紧急,这个任务需要由你来完成。
魔法阵是三角形△的,比如
.............
.x..x....x...
...xxx..x.x..
.......xxxxx.
.............
以上都认为是魔法阵。
(即:三角形三个顶点为(i,j),(i+k,j-k),(i+k,j+k),边上均为’x’。三角形中间的元素没有要求。一个’x’可以同时属于多个魔法阵。单个’x’也算一个魔法阵。)
场地可以认为是一个N×M的矩形,每个位置上为’.’表示没有东西,或’x’表示有魔法阵。
Input
第一行是一个整数T(T<=10),表示共有T组测试数据。
每组数据的第一行包含两个整数N M(N,M≤1000),表示矩阵大小。
接下来N行 ,每行M个字母为‘.’或者‘x’。
Output
每组数据输出独占一行,输出格式为”Case #x: y”,x从1开始,表示数据组号,y表示相应的魔法阵的数量。
Sample Input
13 3.x.xxx...
Sample Output
Case #1: 5
f[i][j]代表i,j右上角延伸出去有多少个x,r[i][j]代表i,j左上角延伸出去,sum[i][j]代表左边延伸出去.均包含自身.
一行一行讨论,每次都假设i,j这个点作为右下角企图构成三角形
首先考虑他的左下角最远能是多少,显然这个值l=max(j-(r[i][j]-1)*2,j-(sum[i][j]-1)).
问题到了这一步,就是求每一个x∈[l,j],满足x+(f[i][x]-1)*2>=j的x的个数
那么可以用主席树解决这个问题,一开始建树就以j+(f[i][j]-1)*2为权值建树,问题可以转化为求[l,j]区间 权值j的rank
问题到了这一步,就是求每一个x∈[l,j],满足x+(f[i][x]-1)*2>=j的x的个数
那么可以用主席树解决这个问题,一开始建树就以j+(f[i][j]-1)*2为权值建树,问题可以转化为求[l,j]区间 权值j的rank
我们用树状数组离线求解一样可以解决这个问题,先统计所有的j+f([i][j]-1)*2,并加个标记1,以及所有的询问j,标记2 排序之后 将标记1的点的位置插入树状数组 对标记2的点进行求区间和 显然 因为求解之前插入的点都满足比他大 求出来的区间和就是需要的答案。注意 这里需要分奇偶建立两颗树状数组
#include<stdio.h>
#include<string>
#include<string.h>
#include<math.h>
#include<map>
#include<iostream>
#include<algorithm>
#define ms(a,b) memset(a,b,sizeof(a))
#define INF 1e9
#define eps 1e-8
#define lowbit(x) ((x)&(-x))
using namespace std;
int N,M;
char MAP[1009][1009];
int Left[1009][1009];
int Right[1009][1009];
int sum[1009][1009];
int Tree[2][2009];
struct node{
int x;
int pos;
int l;
int flag;
}data[2009];
int tot;
int Insert(int k,int x)
{
for(int i=x;i<=M;i+=lowbit(i))
Tree[k][i]+=1;
}
int getsum(int k,int x)
{
int res=0;
for(int i=x;i>0;i-=lowbit(i))
res+=Tree[k][i];
return res;
}
bool cmp(node a,node b)
{
if(a.x==b.x)
return a.flag<b.flag;
return a.x>b.x;
}
int main(){
int T;
scanf("%d",&T);
int Case=0;
while(T--){
cin>>N>>M;
int i,j,k;
memset(Left,0,sizeof(Left));
memset(Right,0,sizeof(Right));
memset(sum,0,sizeof(sum));
for(i=1;i<=N;i++)
{
scanf("%s",&MAP[i][1]);
}
for(i=1;i<=N;i++)
{
for(j=1;j<=M;j++)
{
if(MAP[i][j]=='x')
{
Left[i][j]=Left[i-1][j+1]+1;
Right[i][j]=Right[i-1][j-1]+1;
sum[i][j]=sum[i][j-1]+1;
}
}
}
int ans=0;
for(i=1;i<=N;i++)
{
tot=0;
for(j=0;j<=M;j++)
Tree[0][j]=0;
for(j=0;j<=M;j++)
Tree[1][j]=0;
for(j=1;j<=M;j++)
{
if(Left[i][j]!=0)
{
data[++tot].x=j+(Left[i][j]-1)*2;
if(data[tot].x>M)
data[tot].x=M;
data[tot].pos=j;
data[tot].flag=1;
}
if(MAP[i][j]=='x')
{
data[++tot].x=j;
data[tot].l=max(j-(Right[i][j]-1)*2,j-(sum[i][j]-1));
data[tot].pos=j;
data[tot].flag=2;
}
}
sort(data+1,data+1+tot,cmp);
for(k=1;k<=tot;k++)
{
if(data[k].flag==1)
{
Insert(data[k].pos%2,data[k].pos);
}
else
{
ans+=getsum(data[k].pos%2,data[k].x)-getsum(data[k].pos%2,data[k].l-1);
}
}
}
Case++;
printf("Case #%d: %d\n",Case,ans);
}
return 0;
}