제출 #485847

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
485847		arwaeystoamneg	이상적인 도시 (IOI12_city)	C++17	100 / 100	49 ms	8440 KiB

city

// EXPLOSION!
#define _CRT_SECURE_NO_WARNINGS
#include<unordered_set>
#include<unordered_map>
#include<chrono>
#include<iostream>
#include<bits/stdc++.h>
 
using namespace std;
typedef pair<int, int> pii;
typedef long long ll;
typedef pair<ll, ll> pll;
typedef long double ld;
typedef vector<int> vi;
typedef vector<ll> vll;
typedef vector<pair<int, int>> vpi;
typedef vector<pair<ll, ll>> vpll;
 
#define FOR(i,a,b) for (int i = (a); i < (b); ++i)
#define F0R(i,a) FOR(i,0,a)
#define ROF(i,a,b) for (int i = (b)-1; i >= (a); --i)
#define R0F(i,a) ROF(i,0,a)
#define trav(a,x) for (auto& a: x)
 
#define pb push_back
#define mp make_pair
#define rsz resize
#define sz(x) int(x.size())
#define all(x) x.begin(),x.end()
#define f first
#define s second
#define cont continue
#define endl '\n'
//#define ednl '\n'
#define test int testc;cin>>testc;while(testc--)
#define pr(a, b) trav(x,a)cerr << x << b; cerr << endl;
#define message cout << "Hello World" << endl;
const int dx[4] = { 1,0,-1,0 }, dy[4] = { 0,1,0,-1 }; // for every grid problem!!
const ll linf = 4000000000000000000LL;
const ll inf = 1000000007;//998244353    
 
void pv(vi a) { trav(x, a)cout << x << " "; cout << endl; }void pv(vll a) { trav(x, a)cout << x << " "; cout << endl; }void pv(vector<vi>a) {
	F0R(i, sz(a)) { cout << i << endl; pv(a[i]); cout << endl; }
}void pv(vector<vll>a) { F0R(i, sz(a)) { cout << i << endl; pv(a[i]); }cout << endl; }void pv(vector<string>a) { trav(x, a)cout << x << endl; cout << endl; }
void setIO(string s) {
	ios_base::sync_with_stdio(0); cin.tie(0);
#ifdef arwaeystoamneg
	if (sz(s))
	{
		freopen((s + ".in").c_str(), "r", stdin);
		if (s != "test1")
			freopen((s + ".out").c_str(), "w", stdout);
	}
#endif
}
int n;
vpi a;
vector<vi>adj;
vi sizes;
vector<pair<int, pii>>b;
vi c;
int m;
vi sub;
vi parent;
void dfs(int i, int p)
{
	parent[i] = p;
	sub[i] = sizes[i];
	trav(x, adj[i])
	{
		if (x == p)continue;
		dfs(x, i);
		sub[i] += sub[x];
	}
}
ll sum = 0;
ll solve()
{
	sort(all(a));
	vector<vi>x;
	F0R(i, n)
	{
		if (sz(x) <= a[i].f)x.pb(vi());
		x[a[i].f].pb(i);
	}
	c.pb(0);
	F0R(i, sz(x))
	{
		int last = -2;
		trav(j, x[i])
		{
			int y = a[j].s;
			if (last == y - 1)
			{
				b.back().s.s++;
				sizes[sz(b) - 1]++;
			}
			else
			{
				b.pb({ i,{y,y} });
				sizes.pb(1);
			}
			last = y;
		}
		c.pb(sz(b));
	}
	m = sz(b);
	adj.rsz(m);
	F0R(i, sz(x) - 1)
	{
		int j = c[i], k = c[i + 1];
		while (j < c[i + 1] && k < c[i + 2])
		{
			if ((b[j].s.s >= b[k].s.f && b[j].s.s <= b[k].s.s) || (b[k].s.s >= b[j].s.f && b[k].s.s <= b[j].s.s)) { adj[j].pb(k); adj[k].pb(j); }
			if (b[j].s.s < b[k].s.s)j++;
			else k++;
		}
	}
	parent.rsz(m);
	sub.rsz(m);
	dfs(0, -1);
	ll ans = 0;
	FOR(i, 1, m)
	{
		ans += (sub[0] - sub[i]) * ((ll)sub[i]);
	}
	return ans;
}
int DistanceSum(int N, int* X, int* Y)
{
	n = N;
	a.rsz(n);
	int mx = inf, my = inf;
	F0R(i, n)a[i] = { X[i],Y[i] }, mx = min(mx, X[i]), my = min(my, Y[i]);
	trav(x, a)x.f -= mx, x.s -= my;
	ll ans = 0;
	ans = solve();
	trav(x, a)swap(x.f, x.s);
	adj.clear();
	sizes.clear();
	b.clear();
	c.clear();
	m = 0;
	sub.clear();
	parent.clear();
	ans += solve();
	return ans % 1000000000;
 
}
#ifdef arwaeystoamneg
int main()
{
	setIO("test1");
	
}
#endif

• Subtask 1: 11 / 11
• Subtask 2: 21 / 21
• Subtask 3: 23 / 23
• Subtask 4: 45 / 45