宿舍访问问题（优先队列申请）

Question

问题：MAIT的院长要去参观MAIT的宿舍。如您所知，他是一个非常忙碌的人，因此他决定只访问第一个“K”最近的旅馆。旅馆位于二维平面上。你得到了宿舍的坐标，你必须回答距离原点（Dean's place）最近的第 K 个宿舍的火箭距离

输入格式 输入的第一行包含 Q Total no。的查询和 K 有两种查询：

第一种类型：1 x y 对于第一种类型的查询，您了解了新建宿舍的坐标（x，y）。 second type: 2 查询2nd type，需要输出Kth最近旅馆的Rocket距离到现在。

//院长将永远留在他的地方（原点）。保证在第一个类型 2 查询之前至少有 k 个类型 1 查询。

两点(x2,y2)和(x1,y1)之间的火箭距离定义为(x2 - x1)2 + (y2 - y1)2

约束 1

输出格式 对于类型 2 的每个查询，输出距离 Origin 的第 K 个最近旅馆的 Rocket 距离。//

这是我的代码：

#include <iostream>
#include <queue>
#include <vector>
#include <algorithm>

using namespace std;

class roomno
{
    public:
    int x;
    int y;

    roomno(int x,int y)
    {
        this->x=x;
        this->y=y;
    }
    void print()
    {
        cout<<"location"<<"("<<x<<","<<y<<")"<<endl;
    }

  int distance ()
  {
      return (x*x+y*y);
  }

};

class roomcompare
{
    public:
bool operator() (roomno r1,roomno r2)
{
    return r1.distance()>r2.distance();
}
};


int main() 
{   
    int x[1000]},y[1000];
    int l,k=0;

   priority_queue<roomno,vector<roomno>,roomcompare> pq;
     int n,i,j;
   cin>>n>>l;
     //cin>>n;
     cin.ignore();
    for( i=0;i<n;i++)
    {
        cin>>x[i];
    }
    cin.ignore();
    for( j=0;j<n;j++)
    {
        cin>>y[j];
    }
    cin.ignore();

    for(i=0;i<n;i++ )
    {
        roomno r1(x[i],y[i]);
        pq.push(r1);
    }
    while(!pq.empty()&&k!=l)
    {   k++;
        roomno r2=pq.top();
        r2.print();
        pq.pop();

    }
    return 0;

}

原始代码链接：https://codeshare.io/2j1bkA

我的代码有什么问题？

Answer 1

请考虑在帖子中添加您面临的问题。我看过你的代码，但我对你的代码无能为力，因为我什至在你的代码中看到了语法错误，不知道是不是问题所在。

如果我没有误解您的问题，std::set 将比优先队列更适合，因为无法删除优先队列（最小堆）中的最大项目。以下代码应该可以工作：

#include <iostream>
#include <set>
using namespace std;
using ll = long long;
multiset<ll> distances;
int main() {
    ll n, k;
    cin >> n >> k;
    for(ll i = 0; i < n; ++i) {
        ll query; 
        cin >> query;
        if (query == 1) {
            ll x, y;
            cin >> x >> y;
            distances.insert(x * x + y * y);
            if (distances.size() > k) {
                distances.erase(--distances.end());
            }
        } else {
            cout << *distances.rbegin() << '\n';
        }
    }
    return 0;
}