洛谷: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;
}

在这里插入图片描述

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

猜你喜欢

转载自blog.csdn.net/weixin_42721412/article/details/108459521
今日推荐