Submission #682336

#	Time	Username	Problem	Language	Result	Execution time	Memory
682336		vjudge1	Road Construction (JOI21_road_construction)	C++17	0 / 100	96 ms	9088 KiB

road_construction

#include <bits/stdc++.h>
#include <ext/pb_ds/assoc_container.hpp>
#include <ext/pb_ds/tree_policy.hpp>
using namespace std;
using namespace __gnu_pbds;

#pragma GCC optimization("g", on)
#pragma GCC optimization("03")
#pragma comment(linker, "/stack:200000000")
#pragma GCC optimize("Ofast")
#pragma GCC optimize("inline")
#pragma GCC optimize("-fgcse,-fgcse-lm")
#pragma GCC optimize("-ftree-pre,-ftree-vrp")
#pragma GCC optimize("-ffast-math")
#pragma GCC optimize("-fipa-sra")
#pragma GCC optimize("-fpeephole2")
#pragma GCC optimize("-fsched-spec")
#pragma GCC optimize("Ofast,no-stack-protector")
#pragma GCC target("sse,sse2,sse3,ssse3,sse4,popcnt,abm,mmx,avx,avx2,tune=native")
#pragma GCC optimize("unroll-loops")

#define aboba ios_base::sync_with_stdio(false); cin.tie(NULL); cout.tie(NULL);
#define br break;
#define sp " "
#define en "\n"
#define pb push_back
#define sz size()
#define bg begin()
#define ed end()
#define in insert
#define ss second
#define ff first
#define rbg rbegin()
#define setp(a) cout << fixed; cout << setprecision(a);
#define all(v) v.begin(), v.end()
 
typedef int ll;
typedef long double ld;
typedef pair<ll, ll> pll;
typedef double db;
typedef tree<
    long long,
    null_type,
    less_equal<long long>,
    rb_tree_tag,
    tree_order_statistics_node_update> orset;

void freopen(string s) { freopen((s + ".in").c_str(), "r", stdin); freopen((s + ".out").c_str(), "w", stdout); }
ll bp(ll x, ll y, ll z) { ll res = 1; while (y) { if (y & 1) { res = (res * x) % z; y--; } x = (x * x) % z; y >>= 1; } return res; }
// C(n, k) = ((fact[n] * bp(fact[k], mod - 2)) % mod * bp(fact[n - k], mod - 2)) % mod;
ll lcm(ll a, ll b) { return (a / __gcd(a, b)) * b; }

const ll N = 3e5 + 11;
const ll inf = 1e18 + 7;
ll tt = 1;
ll x[N], y[N], p[N];
vector<pll> v;
ll n, k;

bool chk(ll x) {
	ll cnt = 0;
	for (int i = 1;i < n;i++) {
		if (p[i + 1] > x) continue;
		ll pos = upper_bound(p + 1 + i, p + n + 1, x + p[i]) - p;
		pos--;
		if (p[pos] - p[i] > x) continue;
		//cout << pos << sp << i << en;
		cnt += pos - i;
	}
	return (cnt >= k);
}

void solve() {
	cin >> n >> k;
    vector<ll> ans;
    v.pb({-inf, -inf});
    for (int i = 1;i <= n;i++) {
    	cin >> x[i] >> y[i];
    	v.pb({x[i], y[i]});
	}
	sort(v.bg, v.ed);
	for (int i = 2;i <= n;i++) {
		p[i] = p[i - 1] + abs(v[i].ff - v[i - 1].ff);
	}
	//cout << chk(3) << en;
	ll l = 1, r = p[n] * n, tm = 0;
	while (l <= r) {
		ll mid = (l + r) / 2;
		if (chk(mid)) {
			r = mid - 1;
			tm = mid;
		} else l = mid + 1;
	}
	//cout << tm << en;
	for (int i = 1;i < n;i++) {
		if (p[i + 1] > tm) continue;
		ll pos = upper_bound(p + 1 + i, p + n + 1, tm + p[i]) - p;
		pos--;
		if (p[pos] - p[i] > tm) continue;
		//cout << pos << sp << i << en;
		for (int j = i + 1;j <= pos;j++) {
			ans.pb(abs(v[i].ff - v[j].ff) + abs(v[i].ss - v[j].ss));
		}
	}
	sort(ans.bg, ans.ed);
	for (int i = 0;i < k;i++) cout << ans[i] << en;
}

int main() {    
    aboba  
    //freopen("numbers");
    //cin >> tt;
    for (int i = 1;i <= tt;i++) {
        solve();
    }
}

Compilation message (stderr)

road_construction.cpp:7: warning: ignoring '#pragma GCC optimization' [-Wunknown-pragmas]
    7 | #pragma GCC optimization("g", on)
      | 
road_construction.cpp:8: warning: ignoring '#pragma GCC optimization' [-Wunknown-pragmas]
    8 | #pragma GCC optimization("03")
      | 
road_construction.cpp:9: warning: ignoring '#pragma comment ' [-Wunknown-pragmas]
    9 | #pragma comment(linker, "/stack:200000000")
      | 
road_construction.cpp:54:21: warning: overflow in conversion from 'double' to 'll' {aka 'int'} changes value from '1.0e+18' to '2147483647' [-Woverflow]
   54 | const ll inf = 1e18 + 7;
      |                ~~~~~^~~
road_construction.cpp: In function 'void freopen(std::string)':
road_construction.cpp:48:33: warning: ignoring return value of 'FILE* freopen(const char*, const char*, FILE*)' declared with attribute 'warn_unused_result' [-Wunused-result]
   48 | void freopen(string s) { freopen((s + ".in").c_str(), "r", stdin); freopen((s + ".out").c_str(), "w", stdout); }
      |                          ~~~~~~~^~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
road_construction.cpp:48:75: warning: ignoring return value of 'FILE* freopen(const char*, const char*, FILE*)' declared with attribute 'warn_unused_result' [-Wunused-result]
   48 | void freopen(string s) { freopen((s + ".in").c_str(), "r", stdin); freopen((s + ".out").c_str(), "w", stdout); }
      |                                                                    ~~~~~~~^~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~

• Subtask 1: 0 / 5
• Subtask 2: 0 / 6
• Subtask 3: 0 / 7
• Subtask 4: 0 / 20
• Subtask 5: 0 / 27
• Subtask 6: 0 / 35