STL练习题 2

Problem B: STL-B

Time Limit: 2 Sec  Memory Limit: 128 MB
Submit: 134  Solved: 20

Description

众所周知，华农美旸旸是一个非常男孩，他非常喜欢解决难题，现在他遇到了一个难题，给你N(N <= 10000)个整数x（x <= 1e18）,然后给你一个数sum（sum <= 1e18）,问你能不能从这N个数中找出两个数x, y满足x + y = sum。如果能输出yes,反之输出no.你能帮助美旸旸解决这个问题吗？

Input

第一行包括一个数T(T <= 100),代表测试数据的组数
接下来T组数据
每组数据第一行包括两个整数N, sum,其含义如上
接下来一行包括N个整数。

Output

对于每组数据，如果满足题意，则输出yes，反之输出no

解答

#include<bits/stdc++.h>
using namespace std;
 
long long x[100005];
map<long long,int> mp;
 
int main()
{
    int T=0,N=0;
    long long sum=0;
    cin>>T;
    for (int i=0;i<T;i++)
    {
        cin>>N>>sum;
        int flag=0;
        for (int i=0;i<N;i++)
        {
            cin>>x[i];
            mp[x[i]]++;
        }
        for (int i=0;i<N;i++)
        {
            mp[x[i]]--;
            long long a=sum-x[i];
           if (mp[a]){flag=1;break;}
           mp[x[i]]++;
        }
        if (flag) cout<<"yes"<<endl;
        else cout<<"no"<<endl;
        mp.clear();
    }
    return 0;
}