Submission #494716

#	Time	Username	Problem	Language	Result	Execution time	Memory
494716		flappybird	Bridges (APIO19_bridges)	C++14	44 / 100	3020 ms	15888 KiB

bridges

#include <bits/stdc++.h>
#include <cassert>
#pragma GCC optimize("O3")
#pragma GCC optimize("Ofast")
#pragma GCC optimize("unroll-loops")
#pragma GCC target("avx,avx2,fma")
using namespace std;
typedef int ll;
typedef pair<ll, ll> pll;
#define MAX 201010
#define MAXS 20
#define INF 1000000000000000001
#define bb ' '
#define ln '\n'
#define Ln '\n'
#define B 800
struct dsu {
	ll p[2][MAX], num[2][MAX];
	vector<ll> chk;
	ll qnum = 0;
	dsu(ll N) {
		ll i;
		for (i = 1; i <= N; i++) p[0][i] = i, num[0][i] = 1;
	}
	ll find(ll v) {
		ll pp;
		if (!p[qnum][v]) pp = p[0][v];
		else pp = p[qnum][v];
		if (pp == v) return v;
		else {
			chk.push_back(v);
			return p[qnum][v] = find(pp);
		}
	}
	ll get(ll v) {
		v = find(v);
		if (!num[qnum][v]) return num[0][v];
		else return num[qnum][v];
	}
	void uni(ll a, ll b) {
		a = find(a);
		b = find(b);
		if (a == b) return;
		if (get(b) < get(a)) swap(a, b);
		ll x = get(b) + get(a);
		p[qnum][a] = b;
		num[qnum][b] = x;
		if (qnum == 1) chk.push_back(b), chk.push_back(a);
	}
	void rb() {
		for (auto x : chk) { num[1][x] = p[1][x] = 0; }
		chk.clear();
		qnum = 0;
	}
	void mark() {
		qnum++;
	}
};
struct query {
	ll t;
	ll b, r, s, w;
	ll ind;
	query(ll t, ll x, ll y, ll ind) :ind(ind) {
		query::t = t;
		if (t == 1) b = x, r = y;
		else s = x, w = y;
	}
	query() {}
};
vector<pll> edge;
ll val[MAX];
vector<query> qv[2000];
query qarr[MAX];
ll ans[MAX];
ll chk[MAX];
ll nchk[MAX];
bool cmp(query& a, query& b) {
	return a.w > b.w;
}
signed main() {
	ios::sync_with_stdio(false), cin.tie(0);
	ll N, M;
	cin >> N >> M;
	ll i;
	edge.resize(M + 1);
	for (i = 1; i <= M; i++) {
		ll a, b;
		cin >> a >> b >> val[i];
		edge[i] = { a, b };
	}
	ll Q;
	cin >> Q;
	for (i = 1; i <= Q; i++) {
		ll a, b, c;
		cin >> a >> b >> c;
		qarr[i] = query(a, b, c, i);
		qv[(i - 1) / B].push_back(qarr[i]);
	}
	ll st, en;
	vector<pll> est;
	for (i = 0; ; i++) {
		ll j;
		est.clear();
		for (j = 1; j <= M; j++) est.push_back({ -val[j], j });
		sort(est.begin(), est.end());
		st = 1 + i * B;
		en = (1 + i) * B;
		en = min(en, Q);
		if (st > en) break;
		for (j = st; j <= en; j++) if (qarr[j].t == 1) chk[qarr[j].b] = 1;
		vector<query> qq;
		for (auto q : qv[i]) if (q.t == 2) qq.push_back(q);
		if (qq.size()) sort(qq.begin(), qq.end(), cmp);
		dsu d(N);
		ll loc = 0;
		for (ll j = 0; j < qq.size(); j++) {
			for (; loc < M; loc++) {
				if (chk[est[loc].second]) continue;
				if (qq[j].w > -est[loc].first) break;
				else d.uni(edge[est[loc].second].first, edge[est[loc].second].second);
			}
			//dsu c = d;
			d.mark();
			ll cnt = 0;
			vector<ll> vv;
			for (ll k = qq[j].ind; k >= st; k--) {
				if (qarr[k].t == 1) {
					if (qarr[k].r >= qq[j].w && !nchk[qarr[k].b]) {
						d.uni(edge[qarr[k].b].first, edge[qarr[k].b].second);
					}
					nchk[qarr[k].b] = 1;
					vv.push_back(qarr[k].b);
				}
			}
			for (ll k = en; k > qq[j].ind; k--) {
				if (qarr[k].t == 1) {
					if (!nchk[qarr[k].b] && val[qarr[k].b] >= qq[j].w) {
						d.uni(edge[qarr[k].b].first, edge[qarr[k].b].second);
						nchk[qarr[k].b] = 1;
						vv.push_back(qarr[k].b);
					}
				}
			}
			for (auto x : vv) nchk[x] = 0;
			vv.clear();
			ans[qq[j].ind] = d.get(qq[j].s);
			d.rb();
			//d = c;
		}
		for (j = st; j <= en; j++) {
			if (qarr[j].t == 1) {
				chk[qarr[j].b] = 0;
				val[qarr[j].b] = qarr[j].r;
				assert(est.size() == M);
			}
		}
	}
	for (i = 1; i <= Q; i++) if (qarr[i].t == 2) cout << ans[i] << ln;
}

Compilation message (stderr)

bridges.cpp: In function 'int main()':
bridges.cpp:116:20: warning: comparison of integer expressions of different signedness: 'll' {aka 'int'} and 'std::vector<query>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
  116 |   for (ll j = 0; j < qq.size(); j++) {
      |                  ~~^~~~~~~~~~~
bridges.cpp:124:7: warning: unused variable 'cnt' [-Wunused-variable]
  124 |    ll cnt = 0;
      |       ^~~
In file included from /usr/include/c++/10/cassert:44,
                 from bridges.cpp:2:
bridges.cpp:154:23: warning: comparison of integer expressions of different signedness: 'std::vector<std::pair<int, int> >::size_type' {aka 'long unsigned int'} and 'll' {aka 'int'} [-Wsign-compare]
  154 |     assert(est.size() == M);
      |            ~~~~~~~~~~~^~~~

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