제출 #1227103

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
1227103		kaiboy	Tug of War (BOI15_tug)	C++20	100 / 100	1031 ms	5616 KiB

tug

#include <algorithm>
#include <iostream>
#include <vector>

using namespace std;

const int  N = 30000;
const int N_ = N << 1;
const int  S = N * 40;

int ii[N_], jj[N_], ww[N_], kk[S + 1];
vector<int> eh[N_]; bool visited[N_];
bool dp_[S + 1], *dp = dp_;
bool dq_[S + 1], *dq = dq_;

int dfs(int h_, int i) {
	visited[i] = true;
	int g_ = -1;
	for (int h : eh[i])
		if (h != h_) {
			int j = i ^ ii[h] ^ jj[h];
			if (visited[j])
				g_ = g_ == -1 || g_ == h ? h : -2;
			else {
				int g = dfs(h, j);
				if (g != -1)
					g_ = g_ == -1 || g_ == g ? g : -2;
			}
		}
	return g_;
}

int dfs_(int h_, int i) {
	int s = 0;
	if (h_ != -1)
		s += ww[h_] * (i & 1 ? -1 : 1);
	for (int h : eh[i])
		if (h != h_ && ww[h]) {
			int j = i ^ ii[h] ^ jj[h];
			s += dfs_(h, j);
		}
	return s;
}

int main() {
	ios_base::sync_with_stdio(false), cin.tie(NULL);
	int n, d_; cin >> n >> d_;
	int n_ = n << 1;
	for (int h = 0; h < n_; h++) {
		int i, j, w; cin >> i >> j >> w, i--, j--;
		ii[h] = i << 1 ^ 0, jj[h] = j << 1 ^ 1, ww[h] = w;
		eh[i << 1 ^ 0].push_back(h);
		eh[j << 1 ^ 1].push_back(h);
	}
	int l_ = -d_, r_ = d_;
	for (int i = 0; i < n_; i++)
		if (!visited[i]) {
			int h = dfs(-1, i);
			if (h < 0) {
				cout << "NO\n";
				return 0;
			}
			int w = ww[h]; ww[h] = 0;
			int s = dfs_(-1, ii[h]) + w;
			int t = dfs_(-1, jj[h]) - w;
			if (s > t)
				swap(s, t);
			l_ -= s, r_ -= s;
			kk[t - s]++;
		}
	dp[0] = true;
	for (int b = 1; b <= r_; b++)
		if (kk[b]) {
			for (int r = 0; r < b; r++)
				for (int k = 0, a = r; a <= r_; a += b) {
					if (dp[a])
						k++;
					dq[a] = k > 0;
					if (a >= b * kk[b] && dp[a - b * kk[b]])
						k--;
				}
			for (int a = 0; a <= r_; a++)
				dp[a] = dq[a];
		}
	for (int a = r_; a >= l_ && a >= 0; a--)
		if (dp[a]) {
			cout << "YES\n";
			return 0;
		}
	cout << "NO\n";
	return 0;
}

• Subtask 1: 18 / 18
• Subtask 2: 30 / 30
• Subtask 3: 23 / 23
• Subtask 4: 29 / 29