9-数对的个数

给一个正整数的序列ai,输出共有多少对数字的和为s.
最多有10000000个数字,数字的最大值为100000(思考如果最大为1e9)如果结果过大,就输出对1e9+7取模之后的结果。


代码1(自己写的):

#include <iostream>
#include <stdio.h>
const int maxn = 100003;
#define mod 1000000007;

using namespace std;
int num[maxn];      //存储某个数值的个数
int main()
{
    long int s,x;   //s为所求数对和,x为数值
    int n;            //n为输入的数值个数
    long sum = 0;       //数对个数
    scanf("%ld%ld",&s,&n);
    for(int i=1;i<=n;i++)
    {
        scanf("%ld",&x);
        num[x]++;
    }
    for(int i=0;i<=s/2;i++)
    {
        if(i == (s-i))
            {
                sum = (sum+(num[i]*(num[i]-1)/2))%mod;
            }
        else
            sum = (sum+num[i]*num[s-i])%mod;
    }
    printf("%ld\n",sum);

    return 0;
}

代码2(西交wrong):

#include <iostream>
#include <cstdio>
#include <cstring>
#include <string>
#include <cmath>
#include <cstdlib>
#include <algorithm>
#include <vector>
#include <queue>
#include <stack>
#include <set>
#include <map>
#define MOD 1000000007
using namespace std;
const int maxn = 100005;
typedef pair<int,int> P;
int main()
{
    int n,s,x;
    long long res=0;
    cin>>n>>s;
    memset(num,0,sizeof(num));
    for(int i=1;i<=n;i++)
    {
        scanf("%d",&x);
        num[x]++; //
    }
    for(int i=0;i<=s/2;i++) //(a,b) a<=b
    {//0 1 2 3   7 6 5 4
        if(i==s-i) res=(res+1ll*num[i]*(num[i]-1)/2)%MOD;
        else res=(res+1ll*num[i]*num[s-i])%MOD;
        //1 1 1 1
    }
    cout<<res<<endl;
    //sort(a+1,a+1+n);
    //O(n)
    //0 0 1 1 1 1 1 2 2 2 2 2 3 3 3 3 3     //先排序,然后首尾指针同时向内扫
      //|   |       |          |
    /*
    for(int i=1;i<=n;i++)                   //暴力解法
    {
        if(a[i]>s) break;
        for(int j=i+1;j<=n;j++) //1 2he 2 1
            if(a[i]+a[j]==s) res++;
    }
    */
    //cout<<res%1000000007<<endl;
   return 0;
}

(感谢西交wrong学长提供以上题目练习)

猜你喜欢

转载自blog.csdn.net/outer_star/article/details/81806759