【2050 思维+贡献方案数】大厦 hdu-6496

大厦

Time Limit: 6000/3000 MS (Java/Others)    Memory Limit: 32768/32768 K (Java/Others)
Total Submission(s): 39    Accepted Submission(s): 15

 

Problem Description

现在就让我们来大胆地畅想2050。我们乘坐着无人驾驶的飞行汽车，驰骋在城市的街头，感受着都市的繁华。

我们看到了一栋高楼大厦，大厦的墙面可以看做一个 W×H 的矩形，我们把它的左下角当成(0,0)，右上角当成(W,H)。上面分布着一些LED灯，这些LED灯与地面呈45度倾斜，并且从矩形的边界延伸到另一边界，把大厦分成了若干个区域。我们想数一下这个图里面存在多少个与地面成45度角的矩形，其中四条边都是LED灯的一部分。

 

Input

第一行一个正整数 T (T≤10) 表示数据组数。
对于每组数据，第一行 W,H,n,m (1≤W,H≤109,0≤n,m≤103) 表示矩形的长和宽，以及两种方向的LED灯的个数。
接下来一行 n 个整数c (1≤c≤W+H−1)，表示这个LED灯可以表示成 x+y=c 的形式，保证 c 两两不同。
接下来一行 m 个整数 c (1−H≤c≤W−1)，表示这个LED灯可以表示成 x−y=c 的形式，保证 c 两两不同。

Output

对于每组数据，输出一个整数表示答案，由于答案可能很大，对 109+7 取模。

 

Sample Input

1 21 12 6 5 4 8 14 20 26 30 -6 -1 2 10 14

 

Sample Output

19

 

#include <bits/stdc++.h>
#define ll long long
using namespace std;
const int maxn=1e3+10;
int a[maxn],b[maxn];
int vis[maxn][maxn];
ll pre[maxn];
const ll mod=1e9+7;
struct node
{
    int l,r;
    int num;
}c[maxn];

int main()
{
    pre[1]=1;
    for(int i=2;i<=1000;i++)
        pre[i]=pre[i-1]+i;
    int T;
    scanf("%d",&T);
    while(T--)
    {
        memset(vis,0,sizeof(vis));
        int w,h,n,m;
        scanf("%d%d%d%d",&w,&h,&n,&m);
        int p=max(n,m);
        for(int i=1;i<=p;i++)
            c[i].l=10000,c[i].r=-1,c[i].num=0;  //初始化
        for(int i=1;i<=n;i++)
            scanf("%d",&a[i]);
        for(int j=1;j<=m;j++)
            scanf("%d",&b[j]);
        sort(a+1,a+n+1);
        sort(b+1,b+m+1);
        for(int i=1;i<=m;i++)
        {
            int flag=0;
            for(int j=1;j<=n;j++)
            {
                double y=(1.00*a[j]-1.00*b[i])/2.00;  //计算坐标
                double x=1.00*a[j]-y;
                if(x>=0&&x<=1.00*w&&y>=0&&y<=1.00*h) //坐标如果在大厦内
                {
                    vis[i][j]=1;
                    if(i>=2&&vis[i-1][j])  //如果可以组成方格
                    {
                        c[i].l=min(c[i].l,j); //得到这行方格的左右界限
                        c[i].r=max(c[i].r,j);
                        if(!flag) flag++; //h个界限，h-1个方格
                        else c[i].num++;
                    }
                }
            }
            //cout<<i<<" "<<c[i].l<<" "<<c[i].r<<" "<<c[i].num<<endl;
        }

        ll ans=0;
        for(int i=1;i<=m;i++)
        {
            if(c[i].l<0||c[i].l>n||c[i].r<0||c[i].r>n) continue;
            ans+=pre[c[i].num]; //本行贡献
            int l=c[i].l;
            int r=c[i].r;
            for(int j=i-1;j>=1;j--)
            {
                if(!c[j].num) break;
                if(c[j].l<0||c[j].l>n||c[j].r<0||c[j].r>n) continue;
                l=max(l,c[j].l);
                r=min(r,c[j].r); //得到每一行重复的左右界限
                int num=r-l;
                if(num&&r>=l)  //计算j-i行的贡献
                    ans+=pre[num];
            }
            ans%=mod;
        }

        printf("%lld\n",ans);
    }
    return 0;
}