제출 #148750

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
148750		Powered By Zigui (#200)	십자가 놓기 (FXCUP4_cross)	C++17	100 / 100	287 ms	19176 KiB

cross

#include "cross.h"
#include <bits/stdc++.h>

using namespace std;

typedef long long ll; 
typedef pair<ll, ll> pll;

long long SelectCross(int K, std::vector<int> I, std::vector<int> O) {
	int N = I.size();
	vector<pll> V;	V.resize(N);
	for(int i = 0 ; i < N ; i++) V[i].first = I[i], V[i].second = O[i];
	sort(V.begin(), V.end());
	reverse(V.begin(), V.end());

	int cnt = 0;
	map<ll, int> MP;
	long long ans = 0;
	for(int i = 0 ; i < N ; i++){
		if(cnt == K) {
			MP.begin()->second--;	cnt--;
			if(MP.begin()->second == 0) MP.erase(MP.begin());
		}

		MP[V[i].second]++;	cnt++;
		if(cnt == K){
			ans = max(ans, V[i].first * V[i].first + (MP.begin()->first - V[i].first) * V[i].first * 2);
		}
	}	
	return ans;
}

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