제출 #447507

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
447507		S2speed	다리 (APIO19_bridges)	C++17	13 / 100	104 ms	11248 KiB

bridges

#include<bits/stdc++.h>

using namespace std;

#define all(x) x.begin() , x.end()
#define sze(x) (ll)(x.size())
typedef long long ll;
typedef pair<ll , ll> pll;
typedef pair<bool , pll> plll;

const ll maxn = 3e5 + 16 , inf = 2e18;

ll n , m , q;
ll v[maxn] , u[maxn] , w[maxn];
vector<ll> adj[maxn];
bitset<maxn> mark;
ll cnt;

void DFS(ll r){
	mark[r] = true;
	cnt++;
	for(auto i : adj[r]){
		if(mark[i]) continue;
		DFS(i);
	}
	return;
}

void sub1(){
	for(ll e = 0 ; e < q ; e++){
		ll t;
		cin>>t; t--;
		if(t){
			cnt = 0;
			for(ll i = 0 ; i < n ; i++){
				mark[i] = false;
				adj[i].clear();
			}
			ll s , r;
			cin>>s>>r; s--;
			for(ll i = 0 ; i < m ; i++){
				if(w[i] >= r){
					adj[v[i]].push_back(u[i]); adj[u[i]].push_back(v[i]);
				}
			}
			DFS(s);
			cout<<cnt<<'\n';
		} else {
			ll b , r;
			cin>>b>>r; b--;
			w[b] = r;
		}
	}
	return;
}

struct segtree {

	ll sz = 1;
	vector<ll> val;

	void init(ll n){
		while(sz < n) sz <<= 1;
		val.assign(sz << 1 , inf);
		return;
	}

	void set(ll i , ll k , ll x , ll lx , ll rx){
		if(rx - lx == 1){
			val[x] = k;
			return;
		}
		ll m = (rx + lx) >> 1 , ln = (x << 1) + 1 , rn = ln + 1;
		if(i < m){
			set(i , k , ln , lx , m);
		} else {
			set(i , k , rn , m , rx);
		}
		val[x] = min(val[ln] , val[rn]);
		return;
	}

	void set(ll i , ll k){
		set(i , k , 0 , 0 , sz);
		return;
	}

	ll cal(ll i , ll k , bool c , ll x , ll lx , ll rx){
		if(c){
			if(rx <= i) return inf;
			if(val[x] >= k) return inf;
			if(rx - lx == 1) return lx;
			ll m = (rx + lx) >> 1 , ln = (x << 1) + 1 , rn = ln + 1;
			ll a = cal(i , k , c , ln , lx , m);
			if(a != inf) return a;
			a = cal(i , k , c , rn , m , rx);
			return a;
		} else {
			if(lx >= i) return -1;
			if(val[x] >= k) return -1;
			if(rx - lx == 1) return lx;
			ll m = (rx + lx) >> 1 , ln = (x << 1) + 1 , rn = ln + 1;
			ll a = cal(i , k , c , rn , m , rx);
			if(a != -1) return a;
			a = cal(i , k , c , ln , lx , m);
			return a;
		}
	}

	ll cal(ll i , ll k , bool c){
		return cal(i , k , c , 0 , 0 , sz);
	}

};

segtree st;

void sub2(){
	st.init(n - 1);
	for(ll i = 0 ; i < n - 1 ; i++){
		st.set(i , w[i]);
	}
	for(ll e = 0 ; e < q ; e++){
		ll t;
		cin>>t; t--;
		if(t){
			ll s , y , l , r;
			cin>>s>>y; s--;
			l = st.cal(s , y , 0);
			r = min(n - 1 , st.cal(s , r , 1));
			cout<<r - l<<'\n';
		} else {
			ll b , r;
			cin>>b>>r; b--;
			st.set(b , r);
		}
	}
	return;
}

int main(){
	ios_base::sync_with_stdio(false); cout.tie(0); cin.tie(0);

	cin>>n>>m;
	for(ll i = 0 ; i < m ; i++){
		cin>>v[i]>>u[i]>>w[i]; v[i]--; u[i]--;
	}
	cin>>q;
	if(n <= 1e3 && m <= 1e3 && q <= 1e4){
		sub1();
		return 0;
	}
	bool ch = true;
	for(ll i = 0 ; i < m ; i++){
		ch &= (v[i] == i);
		ch &= (u[i] == i + 1);
	}
	if(ch){
		sub2();
		return 0;
	}
	return 0;
}

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

bridges.cpp: In function 'void sub2()':
bridges.cpp:94:14: warning: 'r' may be used uninitialized in this function [-Wmaybe-uninitialized]
   94 |    ll a = cal(i , k , c , ln , lx , m);
      |           ~~~^~~~~~~~~~~~~~~~~~~~~~~~~
bridges.cpp:127:19: note: 'r' was declared here
  127 |    ll s , y , l , r;
      |                   ^

• Subtask 1: 13 / 13
• Subtask 2: 0 / 16
• Subtask 3: 0 / 17
• Subtask 4: 0 / 14
• Subtask 5: 0 / 13
• Subtask 6: 0 / 27