1. 程式人生 > >BZOJ 3505 [Cqoi2014]數三角形

BZOJ 3505 [Cqoi2014]數三角形

三角形 () ring space ans cst div ++i n+1

題解:容斥原理,總的三角形減去不符合題意的三角形

不符合題意的三角形即三點共線

設一個端點在(0,0),枚舉另一個端點,用gcd算出線段中間夾著的點

因為相同的線段可以同時計算

所以復雜度O(n^2log)

容斥錯了WA了幾發QWQ

#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
typedef long long Lint;

int n,m;
Lint ans;

int Gcd(int a,int b){
	if(b==0)return a;
	return Gcd(b,a%b);
}

int main(){
	cin>>n>>m;
	ans=1LL*((n+1)*(m+1))*((n+1)*(m+1)-1)*((n+1)*(m+1)-2)/6;
//	cout<<ans<<endl;
	for(int i=1;i<=n;++i){
		for(int j=1;j<=m;++j){
			int tmp=Gcd(i,j)-1;
			ans=ans-1LL*tmp*2*(n-i+1)*(m-j+1);
		}
	}
	for(int i=1;i<=n;++i){
		int tmp=i-1;
		ans=ans-1LL*tmp*(n-i+1)*(m+1);
	}
	for(int i=1;i<=m;++i){
		int tmp=i-1;
		ans=ans-1LL*tmp*(m-i+1)*(n+1);
	}
	cout<<ans<<endl;
	return 0;
}

  

BZOJ 3505 [Cqoi2014]數三角形