Submission #149623

#	Time	Username	Problem	Language	Result	Execution time	Memory
149623		Cafe Maru (#200)	Crosses on the Grid (FXCUP4_cross)	C++17	100 / 100	379 ms	26988 KiB

cross

#include "cross.h"
#include<bits/stdc++.h>
using namespace std;

map<int,vector<int>> oi;

long long SelectCross(int K, std::vector<int> I, std::vector<int> O) {
	int N = I.size();
    for(int i=0;i<N;i++) {
        oi[-O[i]].push_back(I[i]);
    }
    long long ans = 0;
    priority_queue<int> pq;
    for(auto& [z, s] : oi) {
        int x = -z;
        for(int i : s) {
            pq.push(-i);
        }
        while(pq.size() > K) pq.pop();
        if (pq.size() == K) {
            int y = -pq.top();
            long long tmp = y*(2LL*x-y);
            if(tmp > ans) ans = tmp;
        }
    }
	return ans;
}

Compilation message (stderr)

cross.cpp: In function 'long long int SelectCross(int, std::vector<int>, std::vector<int>)':
cross.cpp:19:25: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
         while(pq.size() > K) pq.pop();
               ~~~~~~~~~~^~~
cross.cpp:20:23: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
         if (pq.size() == K) {
             ~~~~~~~~~~^~~~

• Subtask 1: 8 / 8
• Subtask 2: 55 / 55
• Subtask 3: 37 / 37