Submission #1118874

#	Time	Username	Problem	Language	Result	Execution time	Memory
1118874		vjudge1	Aliens (IOI16_aliens)	C++17	25 / 100	137 ms	2812 KiB

aliens

// #pragma comment(linker, "/stack:200000000")
// #pragma GCC optimize("Ofast")
// #pragma GCC optimize("O3,unroll-loops")
// #pragma GCC target("sse,sse2,sse3,ssse3,sse4")
// #pragma GCC target("avx2,bmi,bmi2,popcnt,lzcnt")
#ifdef LOCAL

#else
#include "aliens.h"
#endif

// #define _GLIBCXX_DEBUG
// #define _GLIBCXX_DEBUG_PEDANTIC
 
#include <iomanip>
#include <cassert>
#include <iostream>
#include <vector>
#include <random>
#include <algorithm>
#include <map>
#include <set>
#include <functional>
#include <array>
#include <numeric>
#include <queue>
#include <deque>
#include <bitset>
#include <cmath>
#include <climits>
#include <cstdint>
// #include <ext/pb_ds/assoc_container.hpp> 
// #include <ext/pb_ds/tree_policy.hpp>
// #include <ext/rope>
 
// using namespace __gnu_pbds;
// using namespace __gnu_cxx;
using namespace std;
 
 
const int MOD = 998244353;
const long double PI = 3.141592653589793;
using ll = long long;
const ll INF = 1e18;
 
// #define int ll

// --------> sashko123`s defines:

#define itn int     //Vasya sorry :(
#define p_b push_back
#define fi first
#define se second
#define pii std::pair<int, int>
#define oo LLONG_MAX
#define big INT_MAX
#define elif else if

int input()
{
    int x;
    cin >> x;
    return x;
}

// ----------> end of sashko123`s defines (thank you Vasya <3)

// template<typename K, typename V>
// using hash_table = cc_hash_table<K, V, hash<K>>;

// template<typename T>
// using ordered_set = tree<T,null_type,less<T>,rb_tree_tag,tree_order_statistics_node_update>;

// int n, m, k;
vector<pii> a;
const int N = 600;
const int K = 600;
ll dp[N][K];

ll W(ll l, ll r) {
	if (r - l + 1 >= 0)
		return (r - l + 1) * (r - l + 1);
	else
		return 0;
}

vector<pii> recalc(vector<pii> a, int n) {
	sort(a.begin(), a.end(), [&](pii x, pii y) {
		return tuple{x.second, -x.first} < tuple{y.second, -y.first};
	});

	set<pii> st;
	for (int i = 0; i < n; i++) {
		auto [l, r] = a[i];
		while (!st.empty() && st.rbegin()->first >= l) {
			st.erase(prev(st.end()));
		}
		st.insert({l, r});
	}
	
	return vector(st.begin(), st.end());
}

ll take_photos(int n, int m, int k, vector<int> rr, vector<int> cc) {
	a.resize(n);
	for (int i = 0; i < n; i++) {
		int r = rr[i], c = cc[i];
		if (r > c)
			swap(r, c);
		a[i] = {r, c};
	}

	a = recalc(a, n);
	n = a.size();
	a.insert(a.begin(), {0, 0});

	for (int i = 1; i + 1 <= n; i++) {
		if (a[i].first > a[i + 1].first)
			exit(1);
		if (a[i].second > a[i + 1].second)
			exit(1);
	}

	dp[0][0] = 0;
	
	for (int i = 1; i <= n; i++) {
		dp[i][0] = INF;
		for (int c = 1; c <= k; c++) {
			// cerr << "X " << W(a[1].first, a[i].second) << endl;
			dp[i][c] = dp[0][c - 1] + W(a[1].first, a[i].second);
			for (int t = 1; t < i; t++) {
				ll cur = dp[t][c - 1] + W(a[t + 1].first, a[i].second) - W(a[t + 1].first, a[t].second);
				dp[i][c] = min(dp[i][c], cur);
			}
		}
	}
	return dp[n][k];
}	
 

#ifdef LOCAL
int32_t main(int32_t argc, const char * argv[]) {
    cin.tie(0);
    cout.tie(0);
    ios_base::sync_with_stdio(0);
    // insert code here...

	int n, m, k;
	cin >> n >> m >> k;
	vector<int> rr(n);
	vector<int> cc(n);
	for (int i = 0; i < n; i++)
		cin >> rr[i] >> cc[i];
	cout << take_photos(n, m, k, rr, cc);
    return 0;
}
#endif

• Subtask 1: 4 / 4
• Subtask 2: 12 / 12
• Subtask 3: 9 / 9
• Subtask 4: 0 / 16
• Subtask 5: 0 / 19
• Subtask 6: 0 / 40