欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

洛谷:P3799 妖梦拼木棒(普及/提高-,暴力)

程序员文章站 2024-03-04 11:34:41
...

题目:

洛谷:P3799 妖梦拼木棒(普及/提高-,暴力)

分析: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;
}

洛谷:P3799 妖梦拼木棒(普及/提高-,暴力)

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.相加的两个不相等。

相关标签: 暴力枚举