FOJ有奖月赛-2016年4月（校赛热身赛） D 小茗的魔法阵

 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 

我们用树状数组离线求解一样可以解决这个问题，先统计所有的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;
}