BZOJ3485 : [Baltic2012]peaks

时间:2023-03-08 16:51:32
BZOJ3485 : [Baltic2012]peaks

首先将每个平原缩成一个点,建出图,相邻两个点之间的边权为它们高度的较小值。

用Kruskal算法求出这个图的最大生成树,每次合并两个连通块时新建一个点指向它们,得到一棵有根树。

对于每个点,求出它子树内最高的峰,那么对于每个叶子,在它到根的路径上二分查找即可。

时间复杂度$O(nm\log(nm))$。

#include<cstdio>
#include<algorithm>
using namespace std;
const int N=2005,M=200010;
int n,m,i,j,x,y,z,dx,dy,a[N][N],p[N][N],cb;
int cnt,is[M],h[M],f[M],son[M][2],ma[M],q[M],top,ans;
struct P{int x,y;P(){}P(int _x,int _y){x=_x,y=_y;}}b[M/2],fin[M/2];
inline bool cmp(const P&x,const P&y){return a[x.x][x.y]>a[y.x][y.y];}
inline bool cmp2(const P&a,const P&b){return a.x==b.x?a.y>b.y:a.x>b.x;}
inline void read(int&a){char c;while(!(((c=getchar())>='0')&&(c<='9')));a=c-'0';while(((c=getchar())>='0')&&(c<='9'))(a*=10)+=c-'0';}
void dfs(int x,int y){
p[x][y]=cnt;
for(int i=x-1;i<=x+1;i++)for(int j=y-1;j<=y+1;j++){
if(a[i][j]>a[x][y])is[cnt]=0;
if(a[i][j]==a[x][y]&&!p[i][j])dfs(i,j);
}
}
int F(int x){return f[x]==x?x:f[x]=F(f[x]);}
inline void merge(int x,int y,int z){
if(!x||!y)return;
x=F(x),y=F(y);
if(x==y)return;
h[++cnt]=z,f[cnt]=cnt;
son[cnt][0]=x,son[cnt][1]=y;
f[x]=f[y]=cnt;
}
void dfs2(int x){
if(is[x])ma[x]=h[x];
for(int i=0;i<2;i++){
int y=son[x][i];
if(!y)continue;
dfs2(y);
ma[x]=max(ma[x],ma[y]);
}
}
inline void cal(int o){
int l=1,r=top,t=0,mid;
while(l<=r)if(ma[q[mid=(l+r)>>1]]>o)l=(t=mid)+1;else r=mid-1;
fin[++ans]=P(o,h[q[t]]);
}
void dfs3(int x){
if(is[x])cal(h[x]);
q[++top]=x;
for(int i=0;i<2;i++){
int y=son[x][i];
if(!y)continue;
dfs3(y);
}
top--;
}
int main(){
read(n),read(m);
for(i=1;i<=n;i++)for(j=1;j<=m;j++)read(a[i][j]);
for(i=1;i<=n;i++)for(j=1;j<=m;j++)if(!p[i][j])is[++cnt]=1,h[cnt]=a[i][j],dfs(i,j);
for(i=1;i<=n;i++)for(j=1;j<=m;j++)b[++cb]=P(i,j);
sort(b+1,b+cb+1,cmp);
for(i=1;i<=cnt;i++)f[i]=i;
for(i=1;i<=cb;i++){
x=b[i].x,y=b[i].y,z=p[x][y];
for(dx=x-1;dx<=x+1;dx++)for(dy=y-1;dy<=y+1;dy++)if(a[dx][dy]>=a[x][y])merge(p[dx][dy],z,a[x][y]);
}
dfs2(cnt),dfs3(cnt);
sort(fin+1,fin+ans+1,cmp2);
printf("%d\n",ans);
for(i=1;i<=ans;i++)printf("%d %d\n",fin[i].x,fin[i].y);
return 0;
}