Submission #1051340

# Submission time Handle Problem Language Result Execution time Memory
1051340 2024-08-10T04:49:31 Z pawned Wiring (IOI17_wiring) C++17
0 / 100
165 ms 4304 KB
#pragma GCC optimize("O1,O2,O3,Ofast,unroll-loops")

#include <bits/stdc++.h>
using namespace std;

#define fi first
#define se second
#define pb push_back
typedef long long ll;
typedef pair<ll, ll> ii;
typedef vector<ll> vi;

#include "wiring.h"

ll min_total_length(vector<int> r_g, vector<int> b_g) {
	vi r, b;
	for (int x : r_g)
		r.pb(x);
	for (int x : b_g)
		b.pb(x);
	int N = r.size();
	int M = b.size();
	map<int, ll> dp;
	// dp[i][j]: min cost to connect first i red (exclusive)
	// with first j blue (exclusive)
	dp[0] = 0;
	for (int i = 1; i <= N; i++) {
//		cout<<"at "<<i<<endl;
		map<int, ll> newdp;
		// consider only up to previous blue minus 10, next blue plus 10
		auto it = lower_bound(b.begin(), b.end(), r[i]);
		int lb = max(1, (int)(it - b.begin()) - 10);
		int rb = min(M, (int)(it - b.begin()) + 10);
		for (int j = lb; j <= rb; j++) {	// only consider up to consec
//			cout<<"j: "<<j<<endl;
			newdp[j] = 1e18;
			if (newdp.find(j - 1) != newdp.end())
				newdp[j] = min(newdp[j], newdp[j - 1] + abs(r[i - 1] - b[j - 1]));
			if (dp.find(j) != dp.end())
				newdp[j] = min(newdp[j], dp[j] + abs(r[i - 1] - b[j - 1]));
			if (dp.find(j - 1) != dp.end())
				newdp[j] = min(newdp[j], dp[j - 1] + abs(r[i - 1] - b[j - 1]));
		}
		dp = newdp;
/*		cout<<"dp: "<<endl;
		for (pair<int, ll> p : dp)
			cout<<p.fi<<": "<<p.se<<endl;*/
	}
	return dp[M];
}
# Verdict Execution time Memory Grader output
1 Correct 1 ms 344 KB Output is correct
2 Incorrect 1 ms 348 KB 3rd lines differ - on the 1st token, expected: '14340', found: '0'
3 Halted 0 ms 0 KB -
# Verdict Execution time Memory Grader output
1 Incorrect 0 ms 348 KB 3rd lines differ - on the 1st token, expected: '904', found: '0'
2 Halted 0 ms 0 KB -
# Verdict Execution time Memory Grader output
1 Correct 1 ms 348 KB Output is correct
2 Correct 0 ms 348 KB Output is correct
3 Incorrect 165 ms 4304 KB 3rd lines differ - on the 1st token, expected: '1068938599', found: '0'
4 Halted 0 ms 0 KB -
# Verdict Execution time Memory Grader output
1 Correct 1 ms 600 KB Output is correct
2 Incorrect 156 ms 4164 KB 3rd lines differ - on the 1st token, expected: '373710605', found: '0'
3 Halted 0 ms 0 KB -
# Verdict Execution time Memory Grader output
1 Correct 1 ms 344 KB Output is correct
2 Incorrect 1 ms 348 KB 3rd lines differ - on the 1st token, expected: '14340', found: '0'
3 Halted 0 ms 0 KB -