oj.uz

Submission #532127

# Submission time Handle Problem Language Result Execution time Memory
532127 2022-03-02T06:28:30 Z Haruto810198 Railway Trip 2 (JOI22_ho_t4) C++17
25 / 100
 157 ms 45472 KB 
#include <bits/stdc++.h>

using namespace std;

#define int long long
#define double long double

#define FOR(i, l, r, d) for(int i=(l); i<=(r); i+=(d))
#define szof(x) ((int)(x).size())

#define vi vector<int>
#define pii pair<int, int>

#define F first
#define S second

#define pb push_back
#define eb emplace_back
#define mkp make_pair

const int INF = INT_MAX;
const int LNF = INF*INF;
const int MOD = 1000000007;
const int mod = 998244353;
const double eps = 1e-12;

//#pragma GCC optimize("Ofast")
//#pragma GCC optimize("unroll-loops")

const int MAX = 200010;

struct Edge{
	int from, to;
	Edge (int _from, int _to) : from(_from), to(_to) {}
};

int n, k, m, q;
vector<Edge> edges;
vector<vi> nxt;

bool cmp_by_from(Edge a, Edge b){
	if(a.from == b.from) return (a.to < b.to);
	return (a.from < b.from);
}

void build(){
	
	// remove useless edges
	sort(edges.begin(), edges.end(), cmp_by_from);

	vector<Edge> tmp;
	for(Edge e : edges){
		if(tmp.empty()){
			tmp.pb(e);
		}
		else if(tmp.back().from == e.from){
			tmp.pop_back();
			tmp.pb(e);
		}
		else{
			if(tmp.back().to < e.to){
				tmp.pb(e);
			}
		}
	}

	edges = tmp;
	
	// build binary jumping table
	int sz = szof(edges);
	int ptr = 0;
	
	nxt.resize(MAX);
	for(vi& V : nxt){
		V.resize(20);
	}

	FOR(i, 0, sz-1, 1){
		while(ptr < sz-1 and edges[ptr+1].from <= edges[i].to) ptr++;
		nxt[i][0] = ptr;
	}

	FOR(j, 1, 19, 1){
		FOR(i, 0, sz-1, 1){
			nxt[i][j] = nxt[ nxt[i][j-1] ][j-1];
		}
	}

}

int solve(int qfrom, int qto){

	int sz = szof(edges);
	int ret = 0;
	
	// find first train
	if(qfrom < edges[0].from) return INF;

	int L = 0, R = sz-1, mid;
	while(L < R){
		mid = (L + R + 1) / 2;
		if(edges[mid].from <= qfrom) L = mid;
		else R = mid - 1;
	}
	
	if(edges[mid].to < qfrom) return INF;

	int ptr = L;
	
	//cerr<<"first = "<<ptr<<", ";

	// binary jumping
	for(int j = 19; j >= 0; j--){
		if(edges[ nxt[ptr][j] ].to < qto){
			ptr = nxt[ptr][j];
			ret += (1<<j);
		}
	}

	if(edges[ptr].to < qto){
		ptr = nxt[ptr][0];
		ret++;
	}

	//cerr<<"jump = "<<ret<<endl;

	return ret;
}

signed main(){
	
	ios_base::sync_with_stdio(false);
	cin.tie(0);
	cout.tie(0);
	
	cin>>n>>k;
	cin>>m;
	FOR(i, 1, m, 1){
		int _from, _to;
		cin>>_from>>_to;
		if(_from > _to){
			_from *= -1;
			_to *= -1;
		}
		// <- : *= -1, ->

		edges.pb(Edge(_from, _to));
	}
	
	build();

	/*	
	cerr<<"el : "<<endl;
	for(Edge e : el){
		cerr<<e.from<<" "<<e.to<<endl;
	}
	cerr<<endl;
	cerr<<"er : "<<endl;
	for(Edge e : er){
		cerr<<e.from<<" "<<e.to<<endl;
	}
	cerr<<endl;
	*/

	cin>>q;
	FOR(i, 1, q, 1){
		
		int qfrom, qto;
		cin>>qfrom>>qto;
		if(qfrom > qto){
			qfrom *= -1;
			qto *= -1;
		}
		
		int res = solve(qfrom, qto) + 1;
		if(res > 1e6) res = -1;
		cout<<res<<'\n';
	}
	
	return 0;

}

Compilation message

Main.cpp: In function 'long long int solve(long long int, long long int)':
Main.cpp:106:14: warning: 'mid' may be used uninitialized in this function [-Wmaybe-uninitialized]
  106 |  if(edges[mid].to < qfrom) return INF;
      |              ^

Subtask #1
0 / 8.0

# Verdict Execution time Memory Grader output
1 Incorrect 28 ms 39372 KB Output isn't correct
2 Halted 0 ms 0 KB -

Subtask #2
0 / 8.0

# Verdict Execution time Memory Grader output
1 Incorrect 28 ms 39372 KB Output isn't correct
2 Halted 0 ms 0 KB -

Subtask #3
0 / 11.0

# Verdict Execution time Memory Grader output
1 Incorrect 48 ms 40976 KB Output isn't correct
2 Halted 0 ms 0 KB -

Subtask #4
25.0 / 25.0

# Verdict Execution time Memory Grader output
1 Correct 65 ms 41112 KB Output is correct
2 Correct 99 ms 45472 KB Output is correct
3 Correct 123 ms 43840 KB Output is correct
4 Correct 89 ms 44756 KB Output is correct
5 Correct 89 ms 42796 KB Output is correct
6 Correct 89 ms 42924 KB Output is correct
7 Correct 106 ms 43644 KB Output is correct
8 Correct 113 ms 43516 KB Output is correct

Subtask #5
0 / 35.0

# Verdict Execution time Memory Grader output
1 Correct 92 ms 42708 KB Output is correct
2 Correct 157 ms 42500 KB Output is correct
3 Correct 106 ms 40920 KB Output is correct
4 Correct 65 ms 40036 KB Output is correct
5 Incorrect 51 ms 39712 KB Output isn't correct
6 Halted 0 ms 0 KB -

Subtask #6
0 / 13.0

# Verdict Execution time Memory Grader output
1 Incorrect 28 ms 39372 KB Output isn't correct
2 Halted 0 ms 0 KB -