正題

---

題目大意

n個木棒，求用6個木棒組成正方形的方案總數。

---

解題思路

這個正方形邊有的木棒數兩種可能$3,1,1,1$和$2,2,1,1$。

第一種可以枚舉$1$，然后用$f_x$表示兩根木棒和為x的方案數，用$f_x$加容斥可以計算出這個。

第二種直接暴力枚舉$1$的和考慮$2+2$的不同形式，維護兩個指針就可以計算了。

---

$code$

#include<cstdio> #include<algorithm> #define N 5010 #define A int(1e7)+10 #define ll long long using namespace std; ll ab[N],a[N],num[N],k[N],m,n,ans; ll sum,cnt,l,r,s[A]; int main() {freopen("yist.in","r",stdin);freopen("yist.out","w",stdout);scanf("%lld",&m);for(ll i=1;i<=m;i++)scanf("%lld",&ab[i]);sort(ab+1,ab+1+m);for(ll i=1;i<=m;i++){if(ab[i]>ab[i-1]) a[++n]=ab[i];num[n]++,k[i]=n;}for(ll i=1;i<=n;i++)//2+2+1+1{if(num[i]>=2)//夠兩個1{sum=cnt=0;for(l=1,r=i-1;l<=r;l++)//指針{while(l<=r&&a[l]+a[r]>a[i]) r--;if(a[l]+a[r]!=a[i]||l>r) continue;if(l==r){if(num[l]>=4) cnt+=num[l]*(num[l]-1)*(num[l]-2)*(num[l]-3)/24;cnt+=num[l]*(num[l]-1)/2*sum;}else{if(num[l]>=2&&num[r]>=2)cnt+=num[l]*(num[l]-1)/2*num[r]*(num[r]-1)/2;cnt+=num[l]*num[r]*sum;sum+=num[l]*num[r];}}ans+=cnt*num[i]*(num[i]-1)/2;}}for(ll i=1;i<=m;i++)//3+1+1+1{for(ll j=k[i]+1;j<=n;j++)if(num[j]>=3) ans+=num[j]*(num[j]-1)*(num[j]-2)/6*(s[a[j]-ab[i]]);for(ll j=1;j<i;j++)if(ab[i]+ab[j]<=ab[m]) s[ab[i]+ab[j]]++;}printf("%lld",ans); }

總結

以上是生活随笔為你收集整理的jzoj5123-diyiti【统计,容斥】的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。