題目鏈接:點擊打開鏈接
題目大意:給你一對堆,求是否存在一個四維序列(A,B,C,D)其中A<C,B<D,且A!=C或者B!=D,使A到C的哈夫曼距離等於B到D的哈夫曼距離
解題思路:一開始很頭大,各種不會怎麼想都超時,二分又想不出,最後才發現m的重要性,m表示範圍,也就是有2*m種哈夫曼距離,根據抽屜原理,只要超過2*m必有重複,因此剩下來的情況n最大也就1000,暴搜輕鬆過。
代碼:
#include<iostream>
#include<vector>
#include<cmath>
#include<algorithm>
#include<ctime>
#include "cstdio"
#include "string"
#include "string.h"
#include "map"
#include "bitset"
using namespace std;
struct Node
{
int x, y;
bool friend operator<(Node a, Node b)
{
if (a.x == b.y)
return a.y < b.y;
return a.y < b.y;
}
};
map<Node, int>mp;
vector<Node>node;
int vis[2 * 100000 + 1];
int main()
{
int T,n,m;
scanf("%d", &T);
while (T--)
{
mp.clear();
node.clear();
memset(vis, 0, sizeof(vis));
scanf("%d %d", &n, &m);
long long num = 0;
for (int i = 0;i < n;i++)
{
Node temp;
scanf("%d %d", &temp.x, &temp.y);
if (mp[temp])
continue;
mp[temp] = 1;
node.push_back(temp);
}
num = node.size();
if (num*(num - 1) / 2 > 2 * m)
{
puts("YES");
continue;
}
sort(node.begin(), node.end());
bool flag = false;
for (int i = 0;i < num;i++)
{
for (int j = i + 1;j < num;j++)
{
int dx = abs(node[i].x - node[j].x);
int dy = abs(node[i].y - node[j].y);
//cout << dx << " " << dy << endl;
if (vis[dx + dy])
{
flag = true;
break;
}
vis[dx + dy] = 1;
}
if (flag)
break;
}
if (flag)
puts("YES");
else
puts("NO");
}
return 0;
}