Submission #149274

#TimeUsernameProblemLanguageResultExecution timeMemory
149274샌즈뼈 (#200)Crosses on the Grid (FXCUP4_cross)C++17
100 / 100
124 ms6884 KiB
#include "cross.h"
#include <bits/stdc++.h>
using namespace std;

long long SelectCross(int K, std::vector<int> I, std::vector<int> O) {
	int N = I.size();

  vector<pair<int, int>> v;
  for(int i=0; i<N; i++) v.emplace_back(I[i], O[i]);

  sort(v.rbegin(), v.rend());

  //for(int i=0;i <N; i++) cout << v[i].first << ' '; cout << endl;
  //for(int i=0;i <N; i++) cout << v[i].second << ' '; cout << endl;


  priority_queue<int, vector<int>, greater<int>> s;
  for(int i=0; i<K-1; i++) {
    s.push(v[i].second);
  }

  long long res = 0;
  for(int i=K-1; i<N; i++) {
    int ii = v[i].first;
    int oo = s.size() ? min(s.top(), v[i].second) : v[i].second;

    // cout << oo << ' ' << ii << endl;

    res = max(res, 1LL*oo*oo - 1LL*(oo-ii)*(oo-ii));

    s.push(v[i].second);
    if((int)s.size() > K-1) s.pop();
  }

	return res;
}
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...