洛谷:P3799 妖梦拼木棒(普及/提高-,暴力)
程序员文章站
2024-03-04 11:34:41
...
题目:
分析:4=2根等长+2根相加。
本来想做搜索,但是看它只是选2个何不直接二重循环呢?
T了:
#include<bits/stdc++.h>
using namespace std;
int A[100005];
int cnt[5005];
int m;
int maxx=-1;
int ans=0;//1000000007
int main()
{
cin>>m;
for(int i=0;i<m;i++)
{
cin>>A[i]; cnt[A[i]]++;
maxx=max(maxx,A[i]);
}
sort(A,A+m);
for(int i=0;i<m-1;i++)
for(int j=i+1;j<m;j++)
{
if(A[i]+A[j]>maxx) break;
ans=(ans+cnt[A[i]+A[j]]*(cnt[A[i]+A[j]]-1)/2)%1000000007;
}
cout<<ans;
}
ac代码:
#include<bits/stdc++.h>
using namespace std;
int A[100005];
int cnt[5005];
int m;
int maxx=-1;
long long ans=0;//1000000007
int main()
{
cin>>m;
for(int i=0;i<m;i++)
{
cin>>A[i]; cnt[A[i]]++;
maxx=max(maxx,A[i]);
}
sort(A,A+m);
int c=A[0];
int n=0;
for(int i=1;i<m;i++)
{
if(A[i]==A[n]) continue;
A[n+1]=A[i];
n++;
}
n++;
//选的不一样 且不包含 0
int dd;
if(A[0]==0) dd=1;else dd=0;
for(int i=dd;i<n-1;i++)
for(int j=i+1;j<n;j++)
{
if(A[i]+A[j]>maxx) break;
if(cnt[A[i]+A[j]]<2) continue;
ans=(ans+(long long)cnt[A[i]+A[j]]*(cnt[A[i]+A[j]]-1)/2*cnt[A[i]]*cnt[A[j]])%1000000007;
}
if(dd==0)
{
for(int i=1;i<n;i++)
{
if(cnt[A[i]]<3) continue;
ans=(ans+(long long)cnt[A[i]]*(cnt[A[i]]-1)*(cnt[A[i]]-2)/2/3*cnt[0])%1000000007;
}
}
//选的相加的两根一样,
for(int i=dd;i<n;i++)
{
if(A[i]+A[i]>maxx) break;
if(cnt[A[i]+A[i]]<2) continue;
ans=(ans+(long long)cnt[A[i]+A[i]]*(cnt[A[i]+A[i]]-1)/2*cnt[A[i]]*(cnt[A[i]]-1)/2)%1000000007;
}
cout<<ans;
}
搜索的话觉得太麻烦了,对搜素也加深了理解,搜素达到一定的值的和,难以使用for弄出,搜素使用一定的数目,其实就是for循环的简化而已。
此题思路
1. 统计个数。
2.相同消除。
3.利用组合数
按照0的特殊性,进行分类。
1.有零,显然最多一个,然后相同的选3.
没有零:
1.相加的两个相等。
2.相加的两个不相等。
上一篇: File类:目录的遍历
下一篇: 面试题57. 和为s的两个数字