제출 #527724

#제출 시각아이디문제언어결과실행 시간메모리
5277248e7Railway Trip 2 (JOI22_ho_t4)C++17
87 / 100
353 ms39928 KiB
//Challenge: Accepted
#include <bits/stdc++.h>
using namespace std;
#ifdef zisk
void debug(){cout << endl;}
template<class T, class ... U> void debug(T a, U ... b){cout << a << " ", debug(b...);}
template<class T> void pary(T l, T r) {
	while (l != r) cout << *l << " ", l++;
	cout << endl;
}
#else
#define debug(...) 0
#define pary(...) 0
#endif
#define ll long long
#define maxn 100005
#define pii pair<int, int>
#define ff first
#define ss second
#define io ios_base::sync_with_stdio(0);cin.tie(0);
const int inf = 1<<30;
pii ed[200005];
int rig[maxn], lef[maxn];
vector<pii> el[maxn], er[maxn];

struct segtree{
	int type = 0;
	pii seg[4*maxn];
	void pull(int cur) {
		if (type) seg[cur] = max(seg[cur*2], seg[cur*2+1]);
		else seg[cur] = min(seg[cur*2], seg[cur*2+1]);
	}
	void init(int cur, int l, int r){
		if (r <= l) return;
		if (r - l == 1) {
			seg[cur] = {(type ? 0 : inf), l}; 
			return;
		}
		int m = (l + r) / 2;
		init(cur*2, l, m), init(cur*2+1, m, r);
		pull(cur);	
	}
	void modify(int cur, int l, int r, int ind, int val) {
		if (r <= l) return;
		if (r - l == 1) {
			seg[cur] = {val, l};
			return;
		}
		int m = (l + r) / 2;
		if (ind < m) modify(cur*2, l, m, ind, val);
		else modify(cur*2+1, m, r, ind, val);
		pull(cur);
	}
	pii query(int cur, int l, int r, int ql, int qr) {
		if (r <= l || ql >= r || qr <= l) return type ? make_pair(0, 0) : make_pair(inf, 0);
		if (ql <= l && qr >= r) return seg[cur];
		int m = (l + r) / 2;
		pii le = query(cur*2, l, m, ql, qr), ri = query(cur*2+1, m, r, ql, qr);
		return type ? max(le, ri) : min(le, ri);
	}
} sl, sr;

int tol[17][maxn], tor[17][maxn];

vector<int> adj[maxn];
int dis[maxn];
int main() {
	io
	int n, k, m;
	cin >> n >> k >> m;
	for (int i = 0;i < m;i++) {
		cin >> ed[i].ff >> ed[i].ss;
		if (ed[i].ss > ed[i].ff) {
			er[min(ed[i].ss-1, ed[i].ff + k - 1)].push_back({ed[i].ss, 1});
			er[ed[i].ff - 1].push_back({ed[i].ss, -1});
		} else {
			el[max(ed[i].ss+1, ed[i].ff - k + 1)].push_back({ed[i].ss, 1});
			el[ed[i].ff + 1].push_back({ed[i].ss, -1});
		}
	}
	multiset<int> se;
	for (int i = 1;i <= n;i++) {
		for (auto p:el[i]) {
			if (p.ss == 1) se.insert(p.ff);	
			else se.erase(se.find(p.ff));
		}
		lef[i] = (se.size() ? *se.begin() : i);
	}
	se.clear();
	for (int i = n;i >= 1;i--) {
		for (auto p:er[i]) {
			if (p.ss == 1) se.insert(p.ff);	
			else se.erase(se.find(p.ff));
		}
		rig[i] = (se.size() ? *prev(se.end()) : i);
	}

	sl.init(1, 1, n+1);
	sr.type = 1;
	sr.init(1, 1, n+1);
	for (int i = 1;i <= n;i++) {
		sl.modify(1, 1, n+1, i, lef[i]);
		sr.modify(1, 1, n+1, i, rig[i]);
	}
	for (int i = 1;i <= n;i++) {
		pii nxt = sr.query(1, 1, n+1, i+1, rig[i]+1);	
		tor[0][i] = nxt.ss;
		adj[i].push_back(nxt.ss);

		nxt = sl.query(1, 1, n+1, i+1, rig[i]+1);	
		adj[i].push_back(nxt.ss);

		nxt = sl.query(1, 1, n+1, lef[i], i);	
		tol[0][i] = nxt.ss;
		adj[i].push_back(nxt.ss);

		nxt = sr.query(1, 1, n+1, lef[i], i);	
		adj[i].push_back(nxt.ss);
	}
	pary(lef + 1, lef + n + 1);	
	pary(rig + 1, rig + n + 1);

	lef[0] = 0;
	rig[0] = n+1;
	pary(tol[0] + 1, tol[0] + n + 1);
	pary(tor[0] + 1, tor[0] + n + 1);

	for (int i = 1;i < 17;i++) {
		for (int j = 1;j <= n;j++) {
			tol[i][j] = tol[i-1][tol[i-1][j]];
			tor[i][j] = tor[i-1][tor[i-1][j]];
		}
	}
	//brute force
	for (int i = 1;i <= n;i++) {
		adj[i].push_back(tol[0][i]);
		adj[i].push_back(tor[0][i]);
	}
	int q;
	cin >> q;
	if (q == 1 || n <= 2000) {
		while (q--) {
			int s, t;
			cin >> s >> t;
			for (int i = 1;i <= n;i++) dis[i] = inf;
			dis[s] = 0;
			queue<int> que;
			que.push(s);	
			int ans = inf;
			while (que.size()) {
				int cur = que.front();que.pop();
				if ((lef[cur] <= t && t <= cur) || (rig[cur] >= t && t > cur)) {
					ans = dis[cur];
					break;
				}
				for (int v:adj[cur]) {
					if (dis[v] > dis[cur] + 1) {
						dis[v] = dis[cur] + 1;
						que.push(v);
					}
				}
			}
			cout << (ans == inf ? -1 : ans+1) << "\n";
		}
	} else {
		while (q--) {
			int s, t;
			cin >> s >> t;
			int ans = 0;
			if (s < t) {
				for (int i = 16;i >= 0;i--) {
					if (rig[tor[i][s]] < t) {
						s = tor[i][s];
						ans += 1<<i;
					}
				}
				//debug(s, rig[s]);
				if (rig[s] >= t) ans++;
				else if (rig[tor[0][s]] >= t && tor[0][s] != 0) ans += 2;
				else ans = -1;
			} else {
				for (int i = 16;i >= 0;i--) {
					if (lef[tol[i][s]] > t) {
						s = tol[i][s];
						ans += 1<<i;
					}
				}
				//debug(s, rig[s]);
				if (lef[s] <= t) ans++;
				else if (lef[tol[0][s]] <= t && tol[0][s] != 0) ans += 2;
				else ans = -1;
			}
			cout << ans << "\n";
			//debug();
		}
	}
	
}

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

Main.cpp: In function 'int main()':
Main.cpp:13:19: warning: statement has no effect [-Wunused-value]
   13 | #define pary(...) 0
      |                   ^
Main.cpp:120:2: note: in expansion of macro 'pary'
  120 |  pary(lef + 1, lef + n + 1);
      |  ^~~~
Main.cpp:13:19: warning: statement has no effect [-Wunused-value]
   13 | #define pary(...) 0
      |                   ^
Main.cpp:121:2: note: in expansion of macro 'pary'
  121 |  pary(rig + 1, rig + n + 1);
      |  ^~~~
Main.cpp:13:19: warning: statement has no effect [-Wunused-value]
   13 | #define pary(...) 0
      |                   ^
Main.cpp:125:2: note: in expansion of macro 'pary'
  125 |  pary(tol[0] + 1, tol[0] + n + 1);
      |  ^~~~
Main.cpp:13:19: warning: statement has no effect [-Wunused-value]
   13 | #define pary(...) 0
      |                   ^
Main.cpp:126:2: note: in expansion of macro 'pary'
  126 |  pary(tor[0] + 1, tor[0] + n + 1);
      |  ^~~~
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...