제출 #1345344

#제출 시각아이디문제언어결과실행 시간메모리
1345344NonozeAliens (IOI16_aliens)C++20
0 / 100
0 ms352 KiB
#include "aliens.h"
/*
*	Author: Nonoze
*	Created: Sunday 29/03/2026
*/
#include <bits/stdc++.h>
using namespace std;

#ifndef DEBUG
	#define dbg(...)
#endif

// #define cout cerr << "OUT: "
#define endl '\n'
#define endlfl '\n' << flush
#define quit(x) return (void)(cout << x << endl)

template<typename T> void read(T& x) { cin >> x; }
template<typename T1, typename T2> void read(pair<T1, T2>& p) { read(p.first), read(p.second); }
template<typename T> void read(vector<T>& v) { for (auto& x : v) read(x); }
template<typename T1, typename T2> void read(T1& x, T2& y) { read(x), read(y); }
template<typename T1, typename T2, typename T3> void read(T1& x, T2& y, T3& z) { read(x), read(y), read(z); }
template<typename T1, typename T2, typename T3, typename T4> void read(T1& x, T2& y, T3& z, T4& zz) { read(x), read(y), read(z), read(zz); }
template<typename T> void print(vector<T>& v) { for (auto& x : v) cout << x << ' '; cout << endl; }

#define sz(x) (int)(x.size())
#define all(x) (x).begin(), (x).end()
#define rall(x) (x).rbegin(), (x).rend()
#define make_unique(v) sort(all(v)), v.erase(unique(all(v)), (v).end())
#define pb push_back
#define mp(a, b) make_pair(a, b)
#define fi first
#define se second
#define cmin(a, b) a = min(a, b)
#define cmax(a, b) a = max(a, b)
#define YES cout << "YES" << endl
#define NO cout << "NO" << endl
#define QYES quit("YES")
#define QNO quit("NO")

#define int long long
#define double long double
const int inf = numeric_limits<int>::max() / 4;
mt19937 rng(chrono::steady_clock::now().time_since_epoch().count());
const int MOD = 1e9+7, LOG=20;

int intersect(pair<int, int> a, pair<int, int> b) {
	if (a.fi>b.fi) swap(a, b);
	if (a.se<b.fi) return 0;
	return (a.se-b.fi+1)*(a.se-b.fi+1);
}

int get_area(pair<int, int> a) {
	return (a.se-a.fi+1)*(a.se-a.fi+1);
}

int get_area(pair<int, int> a, pair<int, int> b) {
	return get_area(a)+get_area(b)-intersect(a, b);
}

int take_photos(signed n, signed m, signed k, vector<signed> r, vector<signed> c) {
	vector<pair<int, int>> a;
	for (int i=0; i<n; i++) a.pb({min(r[i], c[i]), max(r[i], c[i])});
	sort(all(a), [](pair<int, int> x, pair<int, int> y) {
		if (x.fi==y.fi) return x.se>y.se;
		return x.fi<y.fi;
	});
	{
		vector<pair<int, int>> b;
		for (auto& p: a) {
			if (b.empty() || b.back().se<p.se) b.pb(p);
		}
		swap(a, b);
	}
	if (sz(a)<=k) {
		int res=0;
		for (auto& p: a) res+=get_area(p);
		for (int i=0; i+1<sz(a)-1; i++) res-=intersect(a[i], a[i+1]);
		return res;
	}
	vector<pair<int, int>> cur; cur.pb({0, n-1});
	while (sz(cur)<k) {
		int idx1=-1, idx2=-1, val=-1;
		for (int i=0; i<sz(cur); i++) if (cur[i].se-cur[i].fi>0) {
			for (int j=cur[i].fi; j<cur[i].se; j++) {
				int opt=get_area({a[cur[i].fi].fi, a[cur[i].se].se}) - get_area({a[cur[i].fi].fi, a[j].se}, {a[j+1].fi, a[cur[i].se].se});
				assert(opt>0);
				if (opt>val) {
					val=opt;
					idx1=i, idx2=j;
				}
			}
		}
		assert(idx1!=-1);
		cur.insert(cur.begin()+idx1+1, {idx2+1, cur[idx1].se});
		cur[idx1].se=idx2;
	}
	int res=0;
	for (auto& p: cur) {
		p.fi=a[p.fi].fi, p.se=a[p.se].se;
		res+=get_area({p.fi, p.se});
	}
	for (int i=0; i+1<sz(cur); i++) res-=intersect(cur[i], cur[i+1]);
	return res;
}
#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...