题意:
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+Cj−AiAj,求最小生成树。
题解:
朱刘算法在求最小树形图上是
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);
}