洛谷P3799 妖梦拼木棒

Posted 小时のblog

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了洛谷P3799 妖梦拼木棒相关的知识,希望对你有一定的参考价值。

题目背景

上道题中,妖梦斩了一地的木棒,现在她想要将木棒拼起来。

题目描述

有n根木棒,现在从中选4根,想要组成一个正三角形,问有几种选法?

输入输出格式

输入格式:

 

第一行一个整数n

第二行n个整数,a1,a2,……an(0<ai<=5000),代表每根木棒的长度。

 

输出格式:

 

一行一个整数,对1e9+7取模

 

输入输出样例

输入样例#1:
4 1 1 2 2
输出样例#1:
1

说明

对于30%的数据 N<=5000

对于100%的数据 N<=100000

by-szc

 

题解:枚举+桶+组合数

代码:

 

#include<iostream>
#include<cstdio>
#include<cstring>
const int mod=1e9+7;
using namespace std;
int ans,n,x,mx,sum[100009];
int c(int x){
    return (1LL*x*(x-1)/2)%mod;
}
int main(){
    scanf("%d",&n);
    for(int i=1;i<=n;i++){
        scanf("%d",&x);mx=max(mx,x);sum[x]++;
    }
    for(int i=1;i<=mx;i++){
        for(int j=i;j<=mx;j++){
            if(i+j>mx)break;
            if(i!=j&&sum[i]&&sum[j]&&sum[i+j]>=2)
            ans=(ans%mod+1LL*sum[i]*sum[j]*c(sum[i+j])%mod)%mod;
            else if(i==j)
            ans=(ans%mod+1LL*c(sum[i])*c(sum[i+j])%mod)%mod;
        }
    }
    cout<<ans%mod<<endl;
    return 0;
}

 

以上是关于洛谷P3799 妖梦拼木棒的主要内容,如果未能解决你的问题,请参考以下文章

luogu P3799 妖梦拼木棒

[luogu p3799] 妖梦拼木棒

luoguP3799 妖梦拼木棒 [组合数学]

luogu3799 妖梦拼木棒

AC日记——妖梦斩木棒 洛谷 P3797

洛谷P3797 妖梦斩木棒