답안 #50265

# 제출 시각 아이디 문제 언어 결과 실행 시간 메모리
50265 2018-06-08T20:23:02 Z Bruteforceman 전선 연결 (IOI17_wiring) C++11
0 / 100
20 ms 19236 KB
#include "bits/stdc++.h"
#include "wiring.h"
using namespace std;
const long long inf = 1e15;

vector <long long> dp[200010];
vector <int> group[200010];
vector <long long> first[200010], last[200010];
typedef pair <int, int> pii;

long long min_total_length(std::vector<int> r, std::vector<int> b) {
	int g = 0;
	int n = r.size();
	int m = b.size();
	vector <pii> v;
	for(int i = 0; i < n; i++) {
		v.emplace_back(r[i], 0);
	}
	for(int i = 0; i < m; i++) {
		v.emplace_back(b[i], 1);
	}
	sort(v.begin(), v.end());
	int prevc = -1;
	group[0].push_back(-1);
	for(int i = 0; i < n+m; i++) {
		if(prevc == v[i].second) {
			group[g].push_back(v[i].first);
		} else {
			++g;
			prevc = v[i].second;
			group[g].push_back(-1);
			group[g].push_back(v[i].first);
		}
	}
	for(int i = 0; i <= g; i++) {
		int sz = group[i].size() - 1;
		first[i].resize(sz + 1);
		last[i].resize(sz + 1);
		for(int j = 1; j <= sz; j++) {
			first[i][j] = first[i][j - 1] + group[i][j];
			last[i][j] = last[i][j - 1] + group[i][sz - j + 1];
		}
	}
	for(int i = 1; i <= g+1; i++) {
		int sz = group[i - 1].size() - 1;
		dp[i].resize(sz + 1);
		for(int j = 0; j <= sz; j++) {
			dp[i][j] = inf;
		}
	}
	
	for(int i = 0; i <= g; i++) {
		for(int j : group[i]) {
			cout << j << " ";
		}
		cout << endl;
	}
	dp[1][0] = 0;
	for(int i = 1; i <= g; i++) {
		int sz_prev = group[i - 1].size() - 1;
		int sz_cur = group[i].size() - 1;
		for(int j = sz_prev; j > 0; j--) {
			dp[i][j - 1] = min(dp[i][j - 1], dp[i][j] + abs(group[i - 1][sz_prev - j + 1] - group[i][1]));
		}
		for(int j = 0; j <= min(sz_cur, sz_prev); j++) {
			dp[i + 1][sz_cur - j] = min(dp[i + 1][sz_cur - j], dp[i][j] + abs(last[i - 1][j] - first[i][j]));
		}
		for(int j = sz_cur; j > 0; j--) {
			if(i == 1) break;
			dp[i + 1][j - 1] = min(dp[i + 1][j - 1], dp[i + 1][j] + abs(group[i][sz_cur - j + 1] - group[i - 1].back()));
		}
	}
	return dp[g + 1][0];
}
# 결과 실행 시간 메모리 Grader output
1 Incorrect 20 ms 19064 KB secret mismatch
2 Halted 0 ms 0 KB -
# 결과 실행 시간 메모리 Grader output
1 Incorrect 17 ms 19184 KB secret mismatch
2 Halted 0 ms 0 KB -
# 결과 실행 시간 메모리 Grader output
1 Incorrect 17 ms 19236 KB secret mismatch
2 Halted 0 ms 0 KB -
# 결과 실행 시간 메모리 Grader output
1 Incorrect 20 ms 19236 KB secret mismatch
2 Halted 0 ms 0 KB -
# 결과 실행 시간 메모리 Grader output
1 Incorrect 20 ms 19064 KB secret mismatch
2 Halted 0 ms 0 KB -