学习C++从娃娃抓起!记录下洛谷C++学习和备考过程中的题目,记录每一个瞬间。
附上汇总贴:洛谷刷题C++语言 | 汇总_热爱编程的通信人的博客-CSDN博客
【题目描述】
上道题中,妖梦斩了一地的木棒,现在她想要将木棒拼起来。
有 n 根木棒,现在从中选 4 根,想要组成一个正三角形,问有几种选法?
答案对 10^9+7 取模。
【输入】
第一行一个整数 n。
第二行往下 n 行,每行 1 个整数,第 i 个整数 ai 代表第 i 根木棒的长度。
【输出】
一行一个整数代表答案。
【输入样例】
4
1
1
2
2
【输出样例】
1
【代码详解】
#include <bits/stdc++.h>
using namespace std;
int main()
{
long long counts[5005]={0}, n, tmp, ans=0;
cin >> n;
for (int i=1; i<=n; i++) {
cin >> tmp;
counts[tmp]++;
}
for (int i=1; i<=2500; i++) {
for (int j=i; j<=5000-i; j++) {
int k3=counts[i+j], k1=counts[i], k2=counts[j];
if (k3>=2) {
if (i==j && k1>=2) {
tmp = k1*(k1-1)/2 * k3*(k3-1)/2 % 1000000007;
ans += tmp;
}
if (i!=j && k1>=1 && k2>=1) {
tmp = k1 * k2 * k3*(k3-1)/2 % 1000000007;
ans += tmp;
}
}
}
}
ans %= 1000000007;
cout << ans;
return 0;
}
【运行结果】
4
1
1
2
2
1