Submission #1071844

#	Time	Username	Problem	Language	Result	Execution time	Memory
1071844		mickey080929	Cultivation (JOI17_cultivation)	C++17	80 / 100	2032 ms	4860 KiB

cultivation

#include <bits/stdc++.h>
 
using namespace std;
typedef long long ll;
typedef long double ld;
typedef pair<ll,ll> pll;
 
const ll inf = 2e18;
 
ll R, C;
ll N;

ll mxl[310][310], mxr[310][310], mxsum[310][310];
 
ll solve(vector<pll> p, ll H) {
	vector<pll> upd;
	ll j = 0;
	for (ll i=0; i<p.size(); i++) {
		while (j < p.size() && p[j].first + H <= p[i].first) {
			upd.push_back({p[j].first + H, -p[j].second});
			j ++;
		}
		upd.push_back(p[i]);
	}
	while (j < p.size()) {
		upd.push_back({p[j].first + H, -p[j].second});
		j ++;
	}
	vector<array<ll,4>> save;
	multiset<ll> s1, s2;
	ll idx1 = 0, idx2 = -1;
	for (ll i=0; i<upd.size(); i++) {
		if (upd[i].second > 0) idx2 ++;
		if (upd[i].second > 0) {
			if (!s1.empty()) {
				auto rit = s1.lower_bound(upd[i].second);
				if (rit != s1.begin()) {
					auto lit = prev(rit);
					if (rit != s1.end()) {
						s2.erase(s2.find(*rit - *lit));
					}
					s2.insert(upd[i].second - *lit);
				}
				if (rit != s1.end()) {
					s2.insert(*rit - upd[i].second);
				}
			}
			s1.insert(upd[i].second);
		}
		else {
			s1.erase(s1.find(-upd[i].second));
			if (!s1.empty()) {
				auto rit = s1.lower_bound(-upd[i].second);
				if (rit != s1.begin()) {
					auto lit = prev(rit);
					if (rit != s1.end()) {
						s2.insert(*rit - *lit);
					}
					s2.erase(s2.find(-upd[i].second - *lit));
				}
				if (rit != s1.end()) {
					s2.erase(s2.find(*rit - (-upd[i].second)));
				}
			}
		}
		if (i + 1 != upd.size() && upd[i].first == upd[i+1].first) continue;
		while (idx1 < N && p[idx1].first + H <= upd[i].first) idx1 ++;
		if (idx1 > idx2) {
			save.push_back({upd[i].first, inf, inf, inf});
			continue;
		}
		ll l2 = *s1.begin() - 1, r2 = C - *s1.rbegin();
		ll sum2 = l2 + r2;
		if (!s2.empty()) sum2 = *s2.rbegin() - 1;
		save.push_back({upd[i].first, mxsum[idx1][idx2], mxl[idx1][idx2], mxr[idx1][idx2]});
	}
	ll ans = inf;
	deque<ll> dq[3];
	for (ll i=(ll)save.size()-1; i>=0; i--) {
		for (ll j=0; j<3; j++) {
			while (!dq[j].empty() && save[dq[j].back()][j+1] <= save[i][j+1]) {
				dq[j].pop_back();
			}
			while (!dq[j].empty() && save[dq[j].front()][0] > save[i][0] + R - 1) {
				dq[j].pop_front();
			}
			dq[j].push_back(i);
		}
		ans = min(ans, max(save[dq[0].front()][1], save[dq[1].front()][2] + save[dq[2].front()][3]));
	}
	return ans;
}
 
int main() {
    ios_base :: sync_with_stdio(false); cin.tie(NULL);
    cin >> R >> C;
    cin >> N;
    vector<pll> p(N);
    for (ll i=0; i<N; i++) {
    	cin >> p[i].first >> p[i].second;
    }
    sort(p.begin(), p.end());
    for (ll i=0; i<N; i++) {
    	vector<ll> cur;
    	for (ll j=i; j<N; j++) {
    		if (cur.empty() || cur.back() < p[j].second) {
    			cur.push_back(p[j].second);
    		}
    		else {
    			for (ll k=0; k<cur.size(); k++) {
    				if (cur[k] >= p[j].second) {
    					cur.insert(cur.begin() + k, p[j].second);
    					break;
    				}
    			}
    		}
    		sort(cur.begin(), cur.end());
    		ll l = cur[0] - 1, r = C - cur.back();
			ll sum = l + r;
			for (ll k=0; k+1<cur.size(); k++) {
				sum = max(sum, cur[k+1] - cur[k] - 1);
			}
			mxl[i][j] = l; mxr[i][j] = r;
			mxsum[i][j] = sum;
    	}
    }
    vector<ll> cand, U, D;
    for (ll i=0; i<N; i++) {
    	U.push_back(p[i].first - 1);
    	D.push_back(R - p[i].first);
    	for (ll j=i+1; j<N; j++) {
    		cand.push_back(abs(p[i].first - p[j].first));
    	}
    }
    for (auto &u : U) {
    	for (auto &d : D) {
    		cand.push_back(u + d + 1);
    	}
    }
    sort(cand.begin(), cand.end());
    cand.erase(unique(cand.begin(), cand.end()), cand.end());
    ll ans = inf;
    for (auto &H : cand) {
   		if (H == 0) continue;
    	ans = min(ans, H - 1 + solve(p, H));
    }
    /*for (ll i=1; i<=2*R; i++) {
    	cout << i << ' ' << solve(p, i) << '\n';
    	ans = min(ans, i - 1 + solve(p, i));
    }*/
    cout << ans << '\n';
}

Compilation message (stderr)

cultivation.cpp: In function 'll solve(std::vector<std::pair<long long int, long long int> >, ll)':
cultivation.cpp:18:16: warning: comparison of integer expressions of different signedness: 'll' {aka 'long long int'} and 'std::vector<std::pair<long long int, long long int> >::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   18 |  for (ll i=0; i<p.size(); i++) {
      |               ~^~~~~~~~~
cultivation.cpp:19:12: warning: comparison of integer expressions of different signedness: 'll' {aka 'long long int'} and 'std::vector<std::pair<long long int, long long int> >::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   19 |   while (j < p.size() && p[j].first + H <= p[i].first) {
      |          ~~^~~~~~~~~~
cultivation.cpp:25:11: warning: comparison of integer expressions of different signedness: 'll' {aka 'long long int'} and 'std::vector<std::pair<long long int, long long int> >::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   25 |  while (j < p.size()) {
      |         ~~^~~~~~~~~~
cultivation.cpp:32:16: warning: comparison of integer expressions of different signedness: 'll' {aka 'long long int'} and 'std::vector<std::pair<long long int, long long int> >::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   32 |  for (ll i=0; i<upd.size(); i++) {
      |               ~^~~~~~~~~~~
cultivation.cpp:66:13: warning: comparison of integer expressions of different signedness: 'll' {aka 'long long int'} and 'std::vector<std::pair<long long int, long long int> >::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   66 |   if (i + 1 != upd.size() && upd[i].first == upd[i+1].first) continue;
      |       ~~~~~~^~~~~~~~~~~~~
cultivation.cpp:73:6: warning: variable 'sum2' set but not used [-Wunused-but-set-variable]
   73 |   ll sum2 = l2 + r2;
      |      ^~~~
cultivation.cpp: In function 'int main()':
cultivation.cpp:110:22: warning: comparison of integer expressions of different signedness: 'll' {aka 'long long int'} and 'std::vector<long long int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
  110 |        for (ll k=0; k<cur.size(); k++) {
      |                     ~^~~~~~~~~~~
cultivation.cpp:120:20: warning: comparison of integer expressions of different signedness: 'll' {aka 'long long int'} and 'std::vector<long long int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
  120 |    for (ll k=0; k+1<cur.size(); k++) {
      |                 ~~~^~~~~~~~~~~

• Subtask 1: 5 / 5
• Subtask 2: 10 / 10
• Subtask 3: 15 / 15
• Subtask 4: 30 / 30
• Subtask 5: 20 / 20
• Subtask 6: 0 / 20