模和最大 贪心

题意:

有个数组a, 对于每个a[i],你要找到一个a[j] (i != j) ,使得 (a[i] + a[j]) % p最大。

输入:

输入第一行两个整数n,p。

第二行n个整数a[i]。

2 <= n <= 1e6.

2 <= p <= 1e9.

a[i] < p.

输出:

一行输出n个整数,空格分开,最后一个数后面不要有空格。

#include <iostream>
#include <cstring>
#include <cstdio>
#include <algorithm>
#define INF 0x3f3f3f3f
using namespace std;
const int maxn=1e6+10;
int a[maxn],b[maxn];
int main(void){
    int n,MOD;
    cin>>n>>MOD;
    for(int i=1;i<=n;i++){
        cin>>a[i];
        b[i]=a[i];
    }
    sort(b+1,b+1+n);
    for(int i=1;i<=n;i++){
        int pos=lower_bound(b+1,b+1+n,MOD-a[i])-(b+1);//b[pos]代表小于MOD-a[i]中最大的数
        if(pos>0&&b[pos]!=a[i]){
            cout<<b[pos]+a[i];
        }
        else if(pos>0&&b[pos]==a[i]){//题目中限制i!=j
            if(pos>1)
                cout<<b[pos-1]+a[i];
            else
                cout<<(b[n-(b[n]==a[i])]+a[i])%MOD;
        }
        else{//找不到小于MOD-a[i]的数
            cout<<(b[n-(b[n]==a[i])]+a[i])%MOD;
        }
        cout<<" \n"[i==n];
    }
    return 0;
}

猜你喜欢

转载自blog.csdn.net/shadandeajian/article/details/81488720