C. Going Home (抽屉原理、范围、暴力)

题目

抽屉原理

看题目的数据,其中可以发现两个数相加结果最大也只是5e6,根据抽屉原理,只要有5e6+1个数就至少会有两个数相加得到相同的值。那么也就是说只要枚举出来的两个数相加的答案最多有5e6+1个就一定会得到至少两组坐标满足相加相等,直接暴力即可。

#include<iostream>
using namespace std;
typedef long long ll;
const int Max = 5e6 + 5;
int lst[Max];
int ls[Max][2];
int main()
{
    
    
	int n;cin >> n;
	for (int i = 1;i <= n;i++)cin >> lst[i];
	for (int i = 1;i < n;i++)
	{
    
    
		for (int j = i + 1;j <= n;j++)
		{
    
    
			int sum = lst[i] + lst[j];
			if (ls[sum][0] != 0&&ls[sum][1]!=i&&ls[sum][1]!=j&&ls[sum][0]!=i)
			{
    
    
				cout << "YES" << endl;
				cout << ls[sum][0] << " " << ls[sum][1] << " " << i << " " << j;
				return 0;
			}
			else ls[sum][0] = i, ls[sum][1] = j;
		}
	}
	cout << "NO" << endl;
}

猜你喜欢

转载自blog.csdn.net/asbbv/article/details/114905104