基础算法题——[EZEC-3] 排序（特判）

[EZEC-3] 排序

解题思路

①、题意：希望组成一个环，环上不同两个数字，而且每两个相邻的数字的绝对值 k 必须相同的。

②、我们只要选择两个符合条件数字，以数量最少的数字作为环上元素的数量的 1/2。

注意特判 k==0 a[i]==1的情况，题目要求环的数量至少为 2 。

---

#include<bits/stdc++.h>
#define ll long long
using namespace std;
int a[1000001];
bool b;
ll n, k, um=0, mx;
int main()
{
	cin>>n>>k;
	memset(a, 0, sizeof(a));
	for (int i=0;i<n;i++)
	{
		ll x,y;
		cin>>x>>y;
		a[x]+=y;
		mx=max(x,mx);
	}
	for (ll i=k; i<=mx; i++)
	{
		if(a[i]!=0 && a[i-k]!=0){
			if(k==0){
				if(a[i]==1) continue;
				um = max(um,a[i]*i), b=1;
			}
			else
				um = max(um, min(a[i],a[i-k])*(i+i-k)),b=1;
		}
	}
	if (b==0)
		cout<<"NO";
	else	
		cout<<um;
    return 0;
}