fzu 2058 問題轉化 + 二分

題意

A array contain N number, can you tell me how many different pair i,j that satisfy a[i] + a[j] = M and 1<=i < j <=N.

0 < N <= 100000, 0 <= M <= 1000000000, -1000000000 <= a[i] <= 1000000000

思路:

用map記錄會超時

所以想到手寫二分。思考一下其實滿足題目條件的對數和每個數字的位置並沒有關係——排序+二分查找(範圍來確定個數)即可。

code:

#include<cstdio>
#include<iostream>
#include<sstream>
#include<cstring>
#include<algorithm>
#include<vector>
#include<string>
#include<queue>
#include<map>
#include<set>
#include<cmath>
#include<cctype>
#include<cstdlib>
using namespace std;

#define INF 0x3f3f3f3f
#define PI acos(-1.0)
#define mem(a, b) memset(a, b, sizeof(a))
#define mod 1000000007
typedef pair<int,int> pii;
typedef long long LL;
//------------------------------
const int maxn = 200005;

int a[maxn];
int n,m;

int Scan(){
    int f=1, ret=0;
    char c;
    while((c=getchar())==' ' || c=='\n') ;
    if(c=='-') f=-1;
    else if(c!='+') ret+=c-'0';
    while((c=getchar())<='9' && c>='0')
        ret=ret*10+c-'0';
    return ret*f;
}
int main(){
    while(scanf("%d%d",&n,&m) != EOF){
        for(int i = 0; i < n; i++){
            a[i] = Scan();
        }
        sort(a,a+n);
        long long cnt = 0;
        for(int i = 0; i < n; i++){
            int id1 = lower_bound(a+i+1, a+n, m-a[i]) - a;
            int id2 = upper_bound(a+i+1, a+n, m-a[i]) - a;
            cnt += (id2 - id1);
        }
        cout << cnt << endl;
    }
    return 0;
}

一TLE嚇得我都開掛了....T_T
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章