牛客小白月赛#58 C 经典枚举

牛牛

题意为：有n张数值为(1~m)的牌。先选出n-2张张牌，若无论怎样的n-2张牌的总数值sum%m!=0，则符合不要求，最终的价值为0；若有n-2张牌使得sum%m=0那么符合要求，且价值为剩余两张牌的数值和S=(S%m==0 ? m:S)。游戏价值唯一，问卡牌的游戏价值

题解

比较有意思的枚举。

看起来需要枚举两张牌O(1e5 * 1e5)，但其实只需要枚举一张即可。

记录n张牌的总和为sum，枚举的牌为ai，那么为了让n-2张牌合法，即为了让sum(n-2)%m==0，则第二张牌的值为aj=(sum%m ? sum%m : m).

如果此牌存在，那么游戏价值直接算出，如果此牌不存在，那么答案为0。

用map来实现查找牌存不存在。

代码

#include <iostream>
#include <map>

using namespace std;
const int N=2e5+10;

int n,m,a[N];
map<int,int> mp;

void solve() {
    mp.clear();
    
    cin>>n>>m;
    long long sum=0;
    for(int i=0;i<n;i++) {
        cin>>a[i];
        mp[a[i]]++;
        sum+=a[i];
    }
    
    int res=0;
    for(int i=0;i<n;i++) {
        long long t=sum-a[i];
        mp[a[i]]--;//取走一张牌
        
        int x=(t%m==0 ? m:t%m);
        if(mp[x]) {
             res=( (a[i]+x)%m==0 ? m:(a[i]+x)%m );
             break;
        }
        mp[a[i]]++;//第二张牌不存在，那么取出的就不是这张牌，放回
    }
    cout<<res<<'\n';
}

int main() {
    int t; cin>>t;
    while(t--) solve();
    
    return 0;
}