Codeforces #341div2 B.Wet Shark and Bishops(暴力)

题目链接：

http://codeforces.com/contest/621/problem/B

题目大意：

给n个点，如果两个点在同一条对角线上面，那么他们就会相互攻击，现在问总共会有多少次攻击产生。

范围：

n<=200000,1<=x,y<=1000。

思路：

根据组合原理，如果一条对角线上有t个点，那么就是任选2个点来发生一次攻击，所以次数就是C（2，t）=t*(t-1)/2。

方法一：所以我可以记录一个点上面两条对角线的状况，如果某个方向上已经被标记过了，就不去走了，否则就暴力查看这个点所在对角线上有多少点。

方法二：直接遍历所有对角线，计算出总情况。

这样看来方法二比较清楚，不过当时我用的方法一，改了好久的bug= = 

只贴了方法一的代码：

#include<stdio.h>
#include<string.h>
#define ll  __int64
ll sum,tt;
int vis[1005][1005][2],x[200005],y[200005];
void work(int x,int y)
{
  //  printf("%d %d %I64d \n",x,y,sum);
    vis[x][y][0]++;
    vis[x][y][1]++;
   int i;
    int a=x;int b=y;
    tt=1;
    for(i=1;a+i<=1000&&b+i<=1000;i++)
    if(vis[a+i][b+i][0]&&vis[a+i][b+i][0]<2){
            tt++;vis[a+i][b+i][0]++;}
    for(i=1;a-i>=1&&b-i>=1;i++)
    if(vis[a-i][b-i][0]&vis[a-i][b-i][0]<2){
        tt++;vis[a-i][b-i][0]++;}
    sum=sum+tt*(tt-1)/2;
    tt=1;
    for(i=1;a+i<=1000&&b-i>=1;i++)
    if(vis[a+i][b-i][1]&&vis[a+i][b-i][1]<2){tt++;vis[a+i][b-i][1]++;}
    for(i=1;a-i>=1&&b+i<=1000;i++)
    if(vis[a-i][b+i][1]&&vis[a-i][b+i][1]<2){tt++;vis[a-i][b+i][1]++;}
    sum=sum+tt*(tt-1)/2;
}
int main()
{
    int i,j,k,n;
    while(~(scanf("%d",&n)))
    {
        sum=0;
        memset(vis,0,sizeof(vis));
        for(i=1;i<=n;i++)
            {
                scanf("%d%d",&x[i],&y[i]);
            vis[x[i]][y[i]][0]=vis[x[i]][y[i]][1]=1;
            }
        for(i=1;i<=n;i++)
        {
            int a,b;
            a=x[i];b=y[i];
            if(vis[a][b][0]==1||vis[a][b][1]==1)
            work(a,b);
        }
        printf("%I64d\n",sum);
    }
}