제출 #670118

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
670118		SanguineChameleon	Bitaro’s Party (JOI18_bitaro)	C++11	100 / 100	1676 ms	112732 KiB

bitaro

// BEGIN BOILERPLATE CODE
 
#include <bits/stdc++.h>
using namespace std;
 
#ifdef KAMIRULEZ
	const bool kami_loc = true;
	const long long kami_seed = 69420;
#else
	const bool kami_loc = false;
	const long long kami_seed = chrono::steady_clock::now().time_since_epoch().count();
#endif
 
const string kami_fi = "kamirulez.inp";
const string kami_fo = "kamirulez.out";
mt19937_64 kami_gen(kami_seed);
 
long long rand_range(long long rmin, long long rmax) {
	uniform_int_distribution<long long> rdist(rmin, rmax);
	return rdist(kami_gen);
}
 
long double rand_real(long double rmin, long double rmax) {
	uniform_real_distribution<long double> rdist(rmin, rmax);
	return rdist(kami_gen);
}
 
void file_io(string fi, string fo) {
	if (fi != "" && (!kami_loc || fi == kami_fi)) {
		freopen(fi.c_str(), "r", stdin);
	}
	if (fo != "" && (!kami_loc || fo == kami_fo)) {
		freopen(fo.c_str(), "w", stdout);
	}
}
 
void set_up() {
	if (kami_loc) {
		file_io(kami_fi, kami_fo);
	}
	ios_base::sync_with_stdio(0);
	cin.tie(0);
}
 
void just_do_it();
 
void just_exec_it() {
	if (kami_loc) {
		auto pstart = chrono::steady_clock::now();
		just_do_it();
		auto pend = chrono::steady_clock::now();
		long long ptime = chrono::duration_cast<chrono::milliseconds>(pend - pstart).count();
		string bar(50, '=');
		cout << '\n' << bar << '\n';
		cout << "Time: " << ptime << " ms" << '\n';
	}
	else {
		just_do_it();
	}
}
 
int main() {
	set_up();
	just_exec_it();
	return 0;
}
 
// END BOILERPLATE CODE
 
// BEGIN MAIN CODE
 
const int ms = 1e5 + 20;
const int bl = 1e2 + 20;
vector<int> adj1[ms];
vector<int> adj2[ms];
pair<int, int> d[ms][bl];
int sz[ms];
pair<int, int> tmp[bl * 2];
int a[ms];
bool f[ms];
int dp[ms];
 
int solve(int u) {
	if (dp[u] != -2) {
		return dp[u];
	}
	dp[u] = (f[u] ? -1 : 0);
	for (auto v: adj2[u]) {
		if (solve(v) != -1) {
			dp[u] = max(dp[u], solve(v) + 1);
		}
	}
	return dp[u];
}
 
void just_do_it() {
	int n, m, q;
	cin >> n >> m >> q;
	for (int i = 1; i <= n; i++) {
		d[i][0] = {0, i};
		sz[i] = 1;
	}
	for (int i = 0; i < m; i++) {
		int u, v;
		cin >> u >> v;
		if (u > v) {
			swap(u, v);
		}
		adj1[u].emplace_back(v);
		adj2[v].emplace_back(u);
	}
	for (int u = 1; u <= n; u++) {
		for (int i = 0; i < sz[u]; i++) {
			d[u][i].first++;
		}
		for (auto v: adj1[u]) {
			merge(d[u], d[u] + sz[u], d[v], d[v] + sz[v], tmp, greater<pair<int, int>>());
			int z = sz[u] + sz[v];
			sz[v] = 0;
			for (int i = 0; i < z; i++) {
				if (!f[tmp[i].second]) {
					d[v][sz[v]++] = tmp[i];
					if (sz[v] == bl) {
						break;
					}
					f[tmp[i].second] = true;
				}
			}
			for (int i = 0; i < z; i++) {
				f[tmp[i].second] = false;
			}
		}
		for (int i = 0; i < sz[u]; i++) {
			d[u][i].first--;
		}
	}
	for (int i = 0; i < q; i++) {
		int u;
		cin >> u;
		int k;
		cin >> k;
		for (int j = 0; j < k; j++) {
			cin >> a[j];
			f[a[j]] = true;
		}
		if (k < bl) {
			int res = -1;
			for (int j = 0; j < sz[u]; j++) {
				if (!f[d[u][j].second]) {
					res = d[u][j].first;
					break;
				}
			}
			cout << res << '\n';
		}
		else {
			for (int j = 1; j <= n; j++) {
				dp[j] = -2;
			}
			cout << solve(u) << '\n';
		}
		for (int j = 0; j < k; j++) {
			f[a[j]] = false;
		}
	}
}
 
// END MAIN CODE

컴파일 시 표준 에러 (stderr) 메시지

bitaro.cpp: In function 'void file_io(std::string, std::string)':
bitaro.cpp:30:10: warning: ignoring return value of 'FILE* freopen(const char*, const char*, FILE*)' declared with attribute 'warn_unused_result' [-Wunused-result]
   30 |   freopen(fi.c_str(), "r", stdin);
      |   ~~~~~~~^~~~~~~~~~~~~~~~~~~~~~~~
bitaro.cpp:33:10: warning: ignoring return value of 'FILE* freopen(const char*, const char*, FILE*)' declared with attribute 'warn_unused_result' [-Wunused-result]
   33 |   freopen(fo.c_str(), "w", stdout);
      |   ~~~~~~~^~~~~~~~~~~~~~~~~~~~~~~~~

• Subtask 1: 7 / 7
• Subtask 2: 7 / 7
• Subtask 3: 86 / 86