oj.uz

답안 #845329

# 제출 시각 아이디 문제 언어 결과 실행 시간 메모리
845329 2023-09-06T13:11:47 Z vjudge1 Birmingham (COCI20_birmingham) C++11
35 / 70
 371 ms 524288 KB 
#include <bits/stdc++.h>
#define lg(a) (31 - __builtin_clz((a)))
// #define endl ("\n")
#define pb push_back
#define mp make_pair
#define lb lower_bound
#define ub upper_bound
#define vi vector<int>
#define st first
#define nd second
#define all(aa) aa.begin(), aa.end()
#define rall(aa) aa.rbegin(), aa.rend()
#define forn(i, n) for(int i=0;i<n;i++)
#define trav(e, x) for(auto& e:x)
#define until(n, v) (int) (lower_bound(v.begin(), v.end(), n)-v.begin()) //# of elements < n
#define after(n, v) (int) (v.end()-upper_bound(v.begin(), v.end(), n))   //# of elements > n
#define sameas(n, v) (int) (upper_bound(v.begin(), v.end(), n) - lower_bound(v.begin(), v.end(), n)) //# of elements ==n
typedef long long ll;
using namespace std;
/*

*/

int k;
vector<vector<int>> edges;
vector<int> vis;
vector<int> ans;

void bfs(int node, int d, priority_queue<pair<int, int>> next){
	// cout<<"ENTERED BFS AT NODE: "<<node << " DISTANCE: "<< -d<<endl;
	vis[node] = 1;
	ans[node] = -d;
	for(auto e : edges[node]){
		if(vis[e]==0){
			next.push(mp(d-1, e));
		}
	}
	if(next.size()==0) return;
	while(vis[next.top().second] and next.size()!=0) next.pop();
	if(next.size()==0) return;
	pair<int, int> p = next.top();
	next.pop();
	bfs(p.second, p.first, next);
}


void solve(){
	int n, m, q; cin>>n>>m>>q>>k;
	edges.resize(n+1);
	vis.resize(n+1);
	for(auto &e : vis) e = 0;
	ans.resize(n+1);
	for(int i=0;i<q;i++){
		int a; cin>>a;
		edges[n].pb(a-1);
	}
	
	for(int i=0;i<m;i++){
		int a, b; cin>>a>>b;
		edges[a-1].pb(b-1);
		edges[b-1].pb(a-1);
	}
	// for(int i=0;i<=n;i++){
	// 	cout<<endl<<i<<"E ";
	// 	for(auto e : edges[i]) cout<<e<<' ';
	// }
	priority_queue<pair<int,int>> next;
	bfs(n, 1, next);
	// binary search to min a s.t. a(a+1)k/2 >= ans[i];
	for(int i=0;i<n;i++){
		int l = 0, r= n+1, o=0;
		while(l<r){
			int mid = (l+r)/2;
			if(mid*(mid+1)*k/2 < ans[i]){
				l = mid+1;
			}
			else if(mid*(mid+1)*k/2 >= ans[i]){
				o = mid;
				r = mid;
			}
		}
		cout<<o<<' ';
	}


}

int main(){
	int test; 
    // cin >> test;
    test =1;
   	while (test--){
		solve();
    }
}

Subtask #1
5.0 / 5.0

# 결과 실행 시간 메모리 Grader output
1 Correct 1 ms 344 KB Output is correct
2 Correct 1 ms 348 KB Output is correct
3 Correct 1 ms 604 KB Output is correct

Subtask #2
5.0 / 5.0

# 결과 실행 시간 메모리 Grader output
1 Correct 0 ms 348 KB Output is correct
2 Correct 0 ms 348 KB Output is correct
3 Correct 1 ms 856 KB Output is correct

Subtask #3
5.0 / 5.0

# 결과 실행 시간 메모리 Grader output
1 Correct 0 ms 348 KB Output is correct
2 Correct 0 ms 348 KB Output is correct
3 Correct 1 ms 604 KB Output is correct

Subtask #4
5.0 / 5.0

# 결과 실행 시간 메모리 Grader output
1 Correct 0 ms 348 KB Output is correct
2 Correct 1 ms 436 KB Output is correct
3 Correct 1 ms 604 KB Output is correct

Subtask #5
5.0 / 5.0

# 결과 실행 시간 메모리 Grader output
1 Correct 0 ms 348 KB Output is correct
2 Correct 0 ms 348 KB Output is correct
3 Correct 0 ms 440 KB Output is correct
4 Correct 1 ms 604 KB Output is correct

Subtask #6
5.0 / 5.0

# 결과 실행 시간 메모리 Grader output
1 Correct 1 ms 348 KB Output is correct
2 Correct 0 ms 348 KB Output is correct
3 Correct 0 ms 348 KB Output is correct
4 Correct 1 ms 856 KB Output is correct

Subtask #7
5.0 / 5.0

# 결과 실행 시간 메모리 Grader output
1 Correct 1 ms 344 KB Output is correct
2 Correct 1 ms 344 KB Output is correct
3 Correct 0 ms 348 KB Output is correct
4 Correct 1 ms 604 KB Output is correct

Subtask #8
0 / 5.0

# 결과 실행 시간 메모리 Grader output
1 Runtime error 341 ms 524288 KB Execution killed with signal 9
2 Halted 0 ms 0 KB -

Subtask #9
0 / 5.0

# 결과 실행 시간 메모리 Grader output
1 Runtime error 366 ms 524288 KB Execution killed with signal 9
2 Halted 0 ms 0 KB -

Subtask #10
0 / 5.0

# 결과 실행 시간 메모리 Grader output
1 Runtime error 334 ms 524288 KB Execution killed with signal 9
2 Halted 0 ms 0 KB -

Subtask #11
0 / 5.0

# 결과 실행 시간 메모리 Grader output
1 Runtime error 327 ms 524288 KB Execution killed with signal 9
2 Halted 0 ms 0 KB -

Subtask #12
0 / 5.0

# 결과 실행 시간 메모리 Grader output
1 Runtime error 371 ms 524288 KB Execution killed with signal 9
2 Halted 0 ms 0 KB -

Subtask #13
0 / 5.0

# 결과 실행 시간 메모리 Grader output
1 Runtime error 330 ms 524288 KB Execution killed with signal 9
2 Halted 0 ms 0 KB -

Subtask #14
0 / 5.0

# 결과 실행 시간 메모리 Grader output
1 Runtime error 354 ms 524288 KB Execution killed with signal 9
2 Halted 0 ms 0 KB -