Submission #902366

#	Time	Username	Problem	Language	Result	Execution time	Memory
902366		yellowtoad	Bitaro’s Party (JOI18_bitaro)	C++17	100 / 100	1959 ms	414796 KiB

bitaro

#include <iostream>
#include <vector>
#include <math.h>
#include <algorithm>
#define f first
#define s second
using namespace std;

const int LIM = sqrt(1e5)+5;
int n, m, test, hv[100010], dpp[100010], vis[100010];
vector<pair<int,int>> dp[100010];
vector<int> edge[100010], tnp;
vector<pair<int,int>> tmp;

int main() {
	ios::sync_with_stdio(0);
	cin.tie(0); cout.tie(0);
	cin >> n >> m >> test;
	for (int i = 1; i <= m; i++) {
		int u, v;
		cin >> u >> v;
		edge[v].push_back(u);
	}
	for (int i = 1; i <= n; i++) {
		int maxx = -1;
		tnp.push_back(i);
		for (int j = 0; j < edge[i].size(); j++) {
			for (int k = 0; k < dp[edge[i][j]].size(); k++) {
				hv[dp[edge[i][j]][k].s] = max(hv[dp[edge[i][j]][k].s],dp[edge[i][j]][k].f+1);
				tnp.push_back(dp[edge[i][j]][k].s);
			}
		}
		for (int j = 0; j < tnp.size(); j++) {
			if (!vis[tnp[j]]) {
				vis[tnp[j]] = 1;
				tmp.push_back({hv[tnp[j]],tnp[j]});
			}
		}
		sort(tmp.begin(),tmp.end(),greater<pair<int,int>>());
		for (int j = 0; j < min(LIM,(int)tmp.size()); j++) dp[i].push_back(tmp[j]);
		tmp.clear();
		while (tnp.size()) {
			hv[tnp.back()] = vis[tnp.back()] = 0;
			tnp.pop_back();
		}
	}
	while (test--) {
		int u, x;
		cin >> u >> m;
		for (int i = 1; i <= m; i++) {
			cin >> x;
			tnp.push_back(x);
			hv[x] = 1;
		}
		if (m < LIM) {
			for (int i = 0; i < dp[u].size(); i++) {
				if (!hv[dp[u][i].s]) {
					cout << dp[u][i].f << "\n";
					goto skip;
				}
			}
			cout << -1 << "\n";
			skip:;
		} else {
			for (int i = 1; i <= u; i++) {
				if (hv[i]) dpp[i] = -1e9;
				else dpp[i] = 0;
				for (int j = 0; j < edge[i].size(); j++) dpp[i] = max(dpp[i],dpp[edge[i][j]]+1);
			}
			if (dpp[u] < 0) cout << -1 << "\n";
			else cout << dpp[u] << "\n";
		}
		while (tnp.size()) {
			hv[tnp.back()] = 0;
			tnp.pop_back();
		}
	}
}

Compilation message (stderr)

bitaro.cpp: In function 'int main()':
bitaro.cpp:27:21: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   27 |   for (int j = 0; j < edge[i].size(); j++) {
      |                   ~~^~~~~~~~~~~~~~~~
bitaro.cpp:28:22: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<std::pair<int, int> >::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   28 |    for (int k = 0; k < dp[edge[i][j]].size(); k++) {
      |                    ~~^~~~~~~~~~~~~~~~~~~~~~~
bitaro.cpp:33:21: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   33 |   for (int j = 0; j < tnp.size(); j++) {
      |                   ~~^~~~~~~~~~~~
bitaro.cpp:25:7: warning: unused variable 'maxx' [-Wunused-variable]
   25 |   int maxx = -1;
      |       ^~~~
bitaro.cpp:56:22: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<std::pair<int, int> >::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   56 |    for (int i = 0; i < dp[u].size(); i++) {
      |                    ~~^~~~~~~~~~~~~~
bitaro.cpp:68:23: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   68 |     for (int j = 0; j < edge[i].size(); j++) dpp[i] = max(dpp[i],dpp[edge[i][j]]+1);
      |                     ~~^~~~~~~~~~~~~~~~

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