网易互娱AI研究工程师岗位第一题讨论

// 第一题这个堆排序错在哪了？
#include <iostream>
#include <queue>
#include <cmath>
using namespace std;
struct maze {
    int value;
    double dis_with_center;
    maze(int v, double dis) {
        value = v;
        dis_with_center = dis;
    }
    friend bool operator < (maze a, maze b) {
        return a.dis_with_center > b.dis_with_center;
    }
};
priority_queue<maze> que;
int L;
int num[510][510];

int main () {
    int T;
    int M;
    int x, y;
    cin >> T;
    while (T--) {
        cin >> M >> L;
        for (int i = 0; i < M; i++) {
            for (int j = 0; j < M; j++) {
                cin >> num[i][j];
            }
        }
        cin >> x >> y;
        for (int i = 0; i < M; i++) {
            for (int j = 0; j < M; j++) {
                if (num[i][j] > 0) {
                    que.push(maze(num[i][j], sqrt(pow(x - i, 2) + pow(y - j, 2))));
                }
            }
        }
//        priority_queue<maze> tmp = que;
//        while (!tmp.empty()) {
//            cout << tmp.top().dis_with_center << " " << tmp.top().value << endl;
//            tmp.pop();
//        }
        while (!que.empty()) {
            if (L >= que.top().dis_with_center) {
                L += que.top().value;
                que.pop();
            } else {
                break;
            }
        }
        cout << L << endl;
    }
    return 0;
}