【洛谷 P3799】 妖梦拼木棒【组合数学】

---

解题思路

说白了就是选两根相等的木棒，然后再挑两根相加得原来两根木棒的，组合一下即可。

因为 $m$ 在此题等于$2$ ，那么我们可以将其化简，得：$2/n(n-1)$

---

代码

#include<algorithm>
#include<iostream>
#include<cstdio>
#include<cmath>
#include<cstring>
using namespace std;
const long long INF=1000000007;
long long n,a[100010],maxn,ans,x;

int main(){
	scanf("%lld",&n);
	for(int i=1;i<=n;i++)
	{
		scanf("%lld",&x);
		a[x]++;
		maxn=max(maxn,x);
	}	
	for(int i=2;i<=maxn;i++)
	{
		if(a[i]>1)
		{
			for(int j=1;j<=i/2;j++)
			{
				if((i-j)!=j&&a[j]&&a[i-j])
					ans=(ans+(((a[i]*(a[i]-1)/2)*a[j]*a[i-j]))%INF)%INF;
				if((i-j)==j&&a[j]>1)
					ans=(ans+(((a[i]*(a[i]-1)/2)*(a[j]*(a[j]-1)/2)))%INF)%INF;
			}
		}
	}
	printf("%lld",ans);
}