题解:P14292 [JOI2024 预选赛 R2] 卡牌游戏 / Card Game 2

· · 题解

思路1:

首先因为我们只需要判断里头有没有 a_i+3a_i+6 这两个数,因为如果 a_i 确定,那么就只需要找两个定值,我们考虑拿个桶来存每个 a_i,然后枚举每个数,看桶里头有没有另外两个数就行了。

#include<bits/stdc++.h>
using namespace std;
int a[200200];
int n,aa[200200];
int main(){
    cin>>n;
    for(int i=1;i<=n;i++){
        cin>>aa;
        a[aa]++;
    }
    for(int i=1;i<=n;i++){
        if(a[aa[i]+3]!=0&&a[aa[i]+6]!=0){
            cout<<"Yes\n";
            return 0;
        }
    }
    cout<<"No";
    return 0;
}

思路2:

因为如果 a_i 确定,那么就只需要找两个定值,那么我们就可以先排序,然后用二分来确定有没有这个值。

#include<bits/stdc++.h>
using namespace std;
int a[200200];
int n;
int main(){
    cin>>n;
    for(int i=1;i<=n;i++){
        cin>>aa[i];
    }
    sort(aa+1,aa+n+1);
    for(int i=1;i<=n;i++){
        if(binary_search(aa+1,aa+n+1,a[i]+3)&&binary_search(aa+1,aa+n+1,a[i]+6)){
            cout<<"Yes\n";
            return 0;
        }
    }
    cout<<"No";
    return 0;
}