stl练习B

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

Sample Input

3
4 5
1 2 3 6
3 10
1 1 2
5 7
3 3 2 4 1

Sample Output

yes
no
yes

HINT

第一组数据有2 + 3 = 5,第二组数据找不到两个数的和为10,第三组数据有3 + 4 = 7

#include<bits/stdc++.h>
using namespace std;

long long x[10005];
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]]++; //对x[ i ]键的值进行操作
        }
        for (int i=0;i<N;i++)
        {
           mp[x[i]]--;  //避免sum=x+x的特殊情况
           long long y=sum-x[i];
           if (mp[y]){flag=1;break;}
           mp[x[i]]++;
        }
        if (flag) cout<<"yes"<<endl;
        else cout<<"no"<<endl;
        mp.clear();//前后多组数据不产生影响
    }
    return 0;
}



猜你喜欢

转载自blog.csdn.net/qq_41394420/article/details/81056323