[题解][牛客]xiao白月赛58-C-牛牛[二分]

最新推荐文章于 2023-10-14 19:06:54 发布

O2iginal

最新推荐文章于 2023-10-14 19:06:54 发布

阅读量299

点赞数

分类专栏：题解文章标签：其他

本文链接：https://blog.csdn.net/qq_60726667/article/details/127172354

版权

题解专栏收录该内容

26 篇文章 0 订阅

订阅专栏

题目描述

题意

n个1~m的整数中选择n-2个数, 使得选择的数之和为m的倍数

思路

将选择n-2个数转为选择2各剩下的数
枚举第一个数, 二分符合条件的第二个数是否存在

code

#include<iostream>
#include<algorithm>
using namespace std;
typedef long long ll;
const int N = 2e5 + 10;
ll a[N];

void solv()
{
    int n, m;
    cin >> n >> m;
    ll sum = 0;
    ll t;
    for(int i = 0; i < n; i ++)
    {
        cin >> a[i];
        sum += a[i];
    }
    sort(a, a + n);
    
    ll aim = (sum - 1) % m + 1;
    for(int i = 0; i < n - 1; i ++)
    {
        int l = i + 1, r = n - 1;
        ll key = aim - a[i];
        if(a[l] > key) break;
        while(l < r)
        {
            int mid = l + r >> 1;
            if(a[mid] >= key) r = mid;
            else l = mid + 1;;
        }
        if(a[l] == key)
        {
            if(aim % m == 0) cout << m << '\n';
            else cout << aim % m << '\n';
            return;
        }
    }
    
    aim += m;

    for(int i = 0; i < n - 1; i ++)
    {
       
        int l = i + 1, r = n - 1;
        ll key = aim - a[i];
        if(a[l] > key) break;
        while(l < r)
        {
            int mid = l + r >> 1;
            if(a[mid] >= key) r = mid;
            else l = mid + 1;
        }

//         cout << i << ' ' << a[i] << ' ' << key << ' ' << l << ' ' << a[l] << '\n';
        if(a[l] == key)
        {
            if(aim % m == 0) cout << m << '\n';
            else cout << aim % m << '\n';
            return;
        }
    }
    
    cout << 0 << '\n';
    
}

int main()
{
    int T;
    cin >> T;
    while(T --)
    {
        solv();
    }
    
    return 0;
}