BZOJ 1800 [Ahoi2009]fly 飞行棋

题目链接

思路

终于有一道自己想出来的题了，开心。

因为是矩形，一定有直角，所以考虑直径，之后由于矩形对角线是两条直径，所以考虑组合数。

直径有n条，矩形有c(n,2)个。

#include<iostream>
#include<cstdio>
#include<map>
using namespace std;
int res,hf,sum[25],a[25];
map<int,int>mp;
long long fac(int k)
{
    long long ans=1;
    for(int i=1;i<=k;i++)
        ans*=i;
    return ans;
}
long long cb(int n,int m)
{
    return (fac(n)/fac(m))/fac(n-m);
}
int main()
{
    int n;
    scanf("%d",&n);
    for(int i=1;i<=n;i++)
    {
        scanf("%d",a+i);
        sum[i]=sum[i-1]+a[i];
        mp[sum[i]]++;
    }
    if(sum[n]%2!=0)
    {
        printf("0
");
        return 0;
    }
    hf=sum[n]/2;
    for(int i=1;i<=n;i++)
    {
        if(sum[i]>hf)
            break;
        if(mp[sum[i]+hf])
            res++;
    }
    printf("%lld
",cb(res,2));
    return 0;
}