제출 #59709

#제출 시각아이디문제언어결과실행 시간메모리
59709KieranHorganAliens (IOI16_aliens)C++17
60 / 100
2085 ms726276 KiB
#pragma GCC optimize("Ofast")
#include "aliens.h"
#include <bits/stdc++.h>
using namespace std;
 
#define int long long
 
vector<pair<int, int>> intervals;
int cost(int i, int l) {
	int overlap = max(0ll, intervals[l].second-intervals[l+1].first+1);
	return (intervals[i].second-intervals[l+1].first+1)*(intervals[i].second-intervals[l+1].first+1) - overlap*overlap;
}
 
long long take_photos(signed n, signed m, signed k, vector<signed> r, vector<signed> c) {
	for(int i = 0; i < n; i++) {
		if(c[i] >= r[i])
			intervals.push_back({r[i], -c[i]});
		else
			intervals.push_back({c[i], -r[i]});
	}
	sort(intervals.begin(), intervals.end());
 
	auto interStore = intervals;
	intervals.clear();
	intervals.push_back({-1, -1});
	int ma = -1;
	for(auto p: interStore) {
		if(-p.second > ma) {
			ma = -p.second;
			intervals.push_back({p.first, -p.second});
		}
	}
 
 
	n = intervals.size()-1;
	k = min(k, n);
 
	vector<vector<int>> dp(k+1, vector<int>(n+1, 1ll<<60));
	vector<vector<int>> opt(k+1, vector<int>(n+1, 0));
	dp[0][0] = 0;


	queue<pair<int, int>> ranges;
	for(int j = 1; j <= k; j++) {
		ranges.push({1ll, (int)n+1});
		while(!ranges.empty()) {
			int l = ranges.front().first;
			int r = ranges.front().second;
			ranges.pop();
			if(l==r) continue;

			int i = (l+r)/2;
			for(int cur = opt[j][l-1]; cur < (r==n+1?i:opt[j][r]+1); cur++) {
				if(dp[j-1][cur] + cost(i, cur) < dp[j][i]) {
					opt[j][i] = cur;
					dp[j][i] = dp[j-1][cur] + cost(i, cur);
				}
			}
			if(l+1 != r) {
				ranges.push({l, i});
				ranges.push({i+1, r});
			}
		}
	}
 
    return dp[k][n];
}
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...