20190921模拟赛 御神度(最小生成树的另类解法(Borůvka/SOLLIN算法)+CDQ分治)

题意:
n n n个点的图求最小生成树, ( i , j ) (i,j) (i,j)的边权为 C i + C j − A i A j C_i+C_j-A_iA_j Ci+CjAiAj,求最小生成树。

题解:
朱刘算法在求最小树形图上是 O ( V E ) O(VE) O(VE)的。
那么在最小生成树上面呢?
O ( m log ⁡ m ) ! O(m\log m)! O(mlogm)和克鲁斯卡尔一样优秀。。。。。。
每次找到所有点连接的边权最小的边, 将这些边加入生成树, 将连通块缩成一个新点, 继续操作直到只剩一个点。
每次可能会有两个点选择同一条边, 但可以反证法证明不可能形
成三元以上的环, 因此每次迭代, 至少会选出 m 2 \frac m2 2m 条边( m m m 是当前
缩点后的点数) 。 该部分复杂度为 O ( l o g   n ) O(log \ n) O(log n)
缩点前我们需要算出所有点的最短邻边,发现这个式子可以斜率优化,可以前后缀凸包然后 O ( n ) O(n) O(n)求最优决策点。
但是缩了点之后我们需要算出所有点在缩下来的点之外的最短邻边,相当于每个点有一个颜色,对于每个点求异色点到他的最短邻边。
但是求凸包的线性算法需要先按横坐标排序。
那样就无法区分同色点。
考虑CDQ分治计算贡献。
那么我们先按颜色从小到大排序。
S o l v e ( l c , r c , l , r ) Solve(lc,rc,l,r) Solve(lc,rc,l,r)表示当前处理 [ l c , r c ] [lc,rc] [lc,rc]间的颜色,他们在数组的 [ l , r ] [l,r] [l,r]中。
那么每次求出 m i d c = ( l c + r c ) / 2 midc = (lc+rc)/2 midc=(lc+rc)/2,然后前半段的颜色排好序建凸包后对后半段做贡献,后半段建凸包后对前半段做贡献,在这之前先分治下去就可以求出答案,上来的时候对横坐标归并排序就是 O ( n log ⁡ n ) O(n\log n) O(nlogn)

AC Code:

#include<bits/stdc++.h>
#define maxn 100005
#define LL long long
using namespace std;

int n,m,col[maxn],to[maxn],c[maxn],z[maxn],vis[maxn],zcol[maxn];
LL A[maxn],C[maxn],cst[maxn],ans;
inline LL dis(int a,int b){ return A[a]*A[b]-C[a]-C[b]; }
bool cmp(const int &u,const int &v){ return col[u] == col[v] ? A[u] < A[v] : col[u] < col[v]; }
int q[maxn],L,R;

void Solve(int lc,int rc,int l,int r){
	if(lc >= rc || l >= r) return;
	int midc = (lc + rc) >> 1 , mid = r;
	for(;col[mid] != midc;mid--);
	Solve(lc,midc,l,mid) , Solve(midc+1,rc,mid+1,r);
	L=R=0;
	for(int i=l;i<=mid;i++){
		for(;L<R-1 && (A[q[R-1]]-A[q[R-2]])*(C[c[i]]-C[q[R-2]])-(A[c[i]]-A[q[R-2]])*(C[q[R-1]]-C[q[R-2]]) <= 0;R--);
		q[R++] = c[i];
	}
	for(int i=mid+1;i<=r;i++){
		for(;L<R-1 && (A[q[L+1]]-A[q[L]])*A[c[i]]-(C[q[L+1]]-C[q[L]])>=0;L++);
		if(to[col[c[i]]] == 0 || dis(c[i],q[L]) > cst[col[c[i]]])
			cst[col[c[i]]] = dis(c[i],q[L]) , to[col[c[i]]] = col[q[L]];
	}
	L=R=0;
	for(int i=mid+1;i<=r;i++){
		for(;L<R-1 && (A[q[R-1]]-A[q[R-2]])*(C[c[i]]-C[q[R-2]])-(A[c[i]]-A[q[R-2]])*(C[q[R-1]]-C[q[R-2]]) <= 0;R--);
		q[R++] = c[i];
	}
	for(int i=l;i<=mid;i++){
		for(;L<R-1 && (A[q[L+1]]-A[q[L]])*A[c[i]]-(C[q[L+1]]-C[q[L]])>=0;L++);
		if(to[col[c[i]]] == 0 || dis(c[i],q[L]) > cst[col[c[i]]])
			cst[col[c[i]]] = dis(c[i],q[L]) , to[col[c[i]]] = col[q[L]];
	}
	for(int i=l,j=mid+1,k=l;i<=mid || j<=r;){
		if(i<=mid && (j>r || A[c[i]] < A[c[j]])) z[k++] = c[i++];
		else z[k++] = c[j++];
	}
	for(int i=l;i<=r;i++)
		c[i] = z[i];
}
int tim;
void dfs(int u,int cl){
	zcol[u] = cl,vis[u] = tim,ans+=cst[u];
	q[R++] = u;
	if(vis[to[u]] == tim){
		if(zcol[to[u]] == cl){
			LL g = cst[u];
			for(int t = -1;t != to[u];)
				g = min(cst[t = q[--R]] , g);
			ans -= g;
		}
		else zcol[u] = zcol[to[u]];
		return;
	}
	dfs(to[u],cl);
	if(zcol[u] != zcol[to[u]]) zcol[u] = zcol[to[u]];
}

int main(){
	scanf("%d",&n),m=n;
	for(int i=1;i<=n;i++) scanf("%d",&A[i]);
	for(int i=1;i<=n;i++) scanf("%d",&C[i]),col[i]=i,c[i]=i;
	for(tim=1;m>1;tim++){
		sort(c+1,c+1+n,cmp);
		memset(to,0,sizeof to);
		Solve(1,m,1,n);
		int p = 0;
		for(int i=1;i<=m;i++)
			if(vis[i]!=tim){
				dfs(i,++p);
				if(zcol[i] != p) p--;
			}
		for(int i=1;i<=n;i++)
			col[i] = zcol[col[i]];
		m = p;	
	}
	printf("%lld\n",-ans);
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值