Submission #1192645

---

Source Code Expand

#include <cstdio>
#define N 200003
#define IL inline
#define REP(a,b,c) for(a=b;a<=c;a++)
using namespace std;
typedef long long lol;
lol ans;
struct edge{int u,v,w;};
struct Heap{
	int len;edge a[N<<1];
	IL void up(int i){
		int j=i>>1,x=a[i].w;edge y=a[i];
		while(j && x<a[j].w)a[i]=a[j],i=j,j>>=1;
		a[i]=y;
	}
	IL void down(int i){
		int j=i<<1,x=a[i].w;edge y=a[i];
		while(j<=len){
			if(j<len && a[j+1].w<a[j].w)j++;
			if(a[j].w<x)a[i]=a[j],i=j,j<<=1;else break;
		}
		a[i]=y;
	}
	IL void push(edge b){a[++len]=b;up(len);}
	IL edge pop(){edge b=a[1];a[1]=a[len--];down(1);return b;}
}O,H;
int n,q,fa[N];
IL int rd(){
	int res=0;char c;while((c=getchar())<'0'||c>'9');
	while(c>='0'&&c<='9')res=res*10+c-'0',c=getchar();return res;
}
IL int getf(int a){return a==fa[a]?a:fa[a]=getf(fa[a]);}
IL bool link(int a,int b){a=getf(a),b=getf(b);if(a==b)return false;fa[b]=a;return true;}
int main(){
  n=rd(),q=rd();int t,i,u,v,w;edge e;t=n-1;REP(i,1,t)fa[i]=i;
	REP(i,1,q)u=rd(),v=rd(),w=rd(),O.push((edge){u,v,w});
	REP(t,2,n){
		while(1){
			if(O.len &&(!H.len || O.a[1].w<H.a[1].w)){
				e=O.pop();
				H.push((edge){e.u,(e.u+1)%n,e.w+1});
				H.push((edge){e.v,(e.v+1)%n,e.w+2});
				if(link(e.u,e.v)){ans+=e.w;break;}
			}else{
				e=H.a[1];
				if(!link(e.u,e.v))H.pop();
				else{
					ans+=e.w;e.u=e.v,e.v=(e.v+1)%n,e.w+=2,H.a[1]=e,H.down(1);break;
				}
			}
		}
	}
	printf("%lld",ans);
  return 0;
}

Submission Info

Judge Result