bzoj 3505: [Cqoi2014]数三角形

Description

给定一个nxm的网格,请计算三点都在格点上的三角形共有多少个。下图为4x4的网格上的一个三角形。

注意三角形的三点不能共线。

Input

输入一行,包含两个空格分隔的正整数m和n。

Output

输出一个正整数,为所求三角形数量。

solution


给的 n,m 是格子数,实际上点数是n+1,m+1,所以我们先++n,++m  下面说的n,m是点数

我一开始是想着总方案数是 C(n*m,3),然后减去 竖列,横行,对角线 长度len的 C(len,3) 

但是我只考虑斜率为 1/-1 的对角线,在wa了多次后看题解,发现竟然还有其他的对角线...

其实考虑到那种情况,我也做不出来,因为用 组合数 计算 三点共线的三角形 无法去重...

 


 

补集思想我想出来了

但是正解计算不合法三角形是:

枚举不合法三角形中 最左边和最右边的两个端点,计算中间端点的个数,即为当前端点为这样时的不合法三角形个数

计算中间端点有个定(gui)理(lv):

(x1,y1)和(x2,y2)之间的点数(不包括端点)为 gcd( abs(x1-x2),abs(y1-y2) )-1 个

暴力枚举两个端点是O(n^2*m^2*logn)肯定超时

优化是固定左端点为(1,1) (我习惯最上角是(1,1)) 枚举右端点,得到一条线段,然后将这条线段挪动

得到 (n-i+1)*(m-j+1) 个相同的线段 (这个不是定理,可以自己画画找规律)

O(n*m*logn)

最后 ans=C(n*m,3)-sum(不合法三角形个数)

 

 1 #include<cstring>
 2 #include<cstdio>
 3 #include<iostream>
 4 #define ll long long
 5 #define mem(a,b) memset(a,b,sizeof(a))
 6 using namespace std;
 7 const int N=1066;
 8 const int LEN=303;
 9 ll gcd(ll x,ll y){return y==0?x:gcd(y,x%y);}
10 
11 ll n,m;
12 
13 int main(){
14     
15     scanf("%lld%lld",&n,&m);
16     if(n<m)swap(n,m);
17     ++n;++m;
18     
19     ll ans=(ll)(n*m-1)*(ll)(n*m-2)*n*m/6;
20     ll temp;
21     for(int i=1;i<=n;++i)
22       for(int j=1;j<=m;++j)
23       {
24             if(i==1&&j==1)
25               temp=0;
26             else
27               if(i==1)
28                 temp=j-2;
29             else
30               if(j==1)
31                 temp=i-2;
32             else
33                 temp=gcd(i-1,j-1)-1;
34             
35             if(i!=1&&j!=1)
36                 ans-=(temp*(n-i+1)*(m-j+1)*2);
37             else
38               ans-=(temp*(n-i+1)*(m-j+1));
39         }
40     cout<<ans;
41     //while(1);
42     return 0;
43 } 
44     
code

 

posted @ 2017-08-14 17:46  A_LEAF  阅读(151)  评论(1编辑  收藏  举报