提出 #68992860

---

ソースコード 拡げる

#include<iostream>
#include<queue>
#define pii pair<int,int>
#define fi first
#define se second
using namespace std;
const int N=2e5+10;
int n,m,q,f[N],s[N],tu[N],tv[N],tx[N],ty[N],tz[N],id[N],res[N],cx[N],cy[N];
int finds(int x) {return f[x]==x?x:f[x]=finds(f[x]);}
struct nod
{
	priority_queue <pii,vector<pii>,greater<pii>> q,e;
	void ins(pii te) {q.push(te);}
	void era(pii te) {e.push(te);}
	int siz() {return q.size()-e.size();}
	void cle()
	{
		while(e.size())
		{
			if(q.top()==e.top()) q.pop(),e.pop();
			else break;
		}
	}
	pii top() {cle(); return q.top();}
	void pop() {cle(); q.pop();}
}st[N];
int main()
{
	ios::sync_with_stdio(false),cin.tie(0),cout.tie(0);
	cin>>n>>m;
	for(int i=1;i<=n;i++) s[i]=1,id[i]=f[i]=i;
	for(int i=1;i<=m;i++) cin>>tu[i]>>tv[i];
	cin>>q;
	for(int i=1;i<=q;i++)
	{
		cin>>tx[i]>>ty[i]>>tz[i],cx[i]=cy[i]=(tz[i]+1)/2;
		st[tx[i]].ins({cx[i],i}),st[ty[i]].ins({cy[i],i});
	}
	for(int i=1;i<=m;i++)
	{
		int x=finds(tu[i]),y=finds(tv[i]);
		if(x==y) continue;
		if(s[x]>s[y]) swap(x,y);
		s[y]+=s[x],f[x]=y;
		if(st[id[x]].siz()>st[id[y]].siz()) swap(id[x],id[y]);
		while(st[id[x]].siz()) st[id[y]].ins(st[id[x]].top()),st[id[x]].pop();
		while(st[id[y]].siz())
		{
			pii te=st[id[y]].top();
			if(s[y]>=te.fi)
			{
				int ti=te.se,xx=tx[ti],yy=ty[ti],z=tz[ti];
				if(ti>q) {st[id[y]].pop(),res[ti-q]=i; continue;}
				int fx=finds(xx),fy=finds(yy);
				st[id[fx]].era({cx[ti],ti}),st[id[fy]].era({cy[ti],ti});
				if(fx==fy)
				{
					if(s[y]>=z) res[ti]=i;
					else st[id[fy]].ins({z,q+ti});
				}
				else
				{
					if(s[fx]+s[fy]>=z) res[ti]=i;
					else
					{
						int dt=z-(s[fx]+s[fy]);
						cx[ti]=s[fx]+(dt+1)/2,cy[ti]=s[fy]+(dt+1)/2;
						st[id[fx]].ins({cx[ti],ti}),st[id[fy]].ins({cy[ti],ti});
					}
				}
			}
			else break;
		}
	}
	for(int i=1;i<=q;i++) cout<<res[i]<<'\n';
	return 0;
}

提出情報

ジャッジ結果