码迷,mamicode.com
首页 > 其他好文 > 详细

洛谷 P2421 A-B数对(增强版)

时间:2017-10-15 22:41:24      阅读:246      评论:0      收藏:0      [点我收藏+]

标签:blank   string   输出   clu   ref   amp   www.   cst   表示   

题目背景

woshiren在洛谷刷题,感觉第一题:求两数的和(A+B Problem)太无聊了,于是增加了一题:A-B Problem,难倒了一群小朋友,哈哈。

题目描述

给出N 个从小到大排好序的整数,一个差值C,要求在这N个整数中找两个数A 和B,使得A-B=C,问这样的方案有多少种?

例如:N=5,C=2,5 个整数是:2 2 4 8 10。答案是3。具体方案:第3 个数减第1 个数;第3 个数减第2 个数;第5 个数减第4 个数。

输入输出格式

输入格式:

 

第一行2 个正整数:N,C。

第二行N 个整数:已经有序。注意:可能有相同的。

 

输出格式:

 

一个整数,表示该串数中包含的所有满足A-B=C 的数对的方案数。

 

输入输出样例

输入样例#1:
4 1
1 1 2 2
输出样例#1:
4

说明

对于50% 的数据:N 的范围是[1…1,000]。

对于另外50% 的数据:N 的范围是[1…100,000]。

对于100% 的数据:C 的范围是[1…1,000,000,000],N 个整数中每个数的范围是:[0…1,000,000,000]。

思路:二分答案

#include<map>
#include<set>
#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;
set<int>se;
map<int,int>ma;
int n,c,num,ans;
int l,r,mid;
int val[100000100];
int main(){
    scanf("%d%d",&n,&c);
    for(int i=1;i<=n;i++){
        scanf("%d",&r);
        ma[r]++;
        if(se.find(r)==se.end()){
            val[++num]=r;
            se.insert(r);
        }
    }
    for(int i=num;i>=1;i--){
        int z=val[i]-c;
        l=0;r=i;;
        bool bns=0;
        for(int j=1;j<=50;j++){
            mid=(l+r)/2;
            if(z>val[mid])    l=mid+1;
            else if(z<val[mid])    r=mid-1;
            else if(z==val[mid]){
                bns=1;
                break;
            }
        }
        if(bns)    ans+=ma[val[i]]*ma[val[mid]];
    }
    cout<<ans;
}

 

洛谷 P2421 A-B数对(增强版)

标签:blank   string   输出   clu   ref   amp   www.   cst   表示   

原文地址:http://www.cnblogs.com/cangT-Tlan/p/7674391.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!