Codeforces 936E. Iqea

Description

给出一张四连通网格图,其中有 \(n\) 个点是连通的,维护以下两种操作:
1.把某个点变黑
2.给出一个白点,查询离这个白点最近的黑点的距离
题面

Solution

我们把每一行看作一个节点,建立一棵树
然后点分治维护这两个操作即可
实际上就是动态加入黑点,查询离某个点最近的黑点
唯一的区别在于:
走到这个节点之后还会在这个节点所代表的行里面走,设查询的为 \(x\),某个黑点为 \(u\)
要求: \(min(dis[u]+dis[x]+|y_u-y_x|)\)
\(dis[i]\) 表示从 \(i\) 出发走到这个重心所代表的行的最短路,\(y_i\) 表示走最短路碰到这个节点所代表的行的纵坐标
两者都可以从重心 \(bfs\) 求出
然后对于每一个重心维护数状数组查询 \(min(dis[u]+dis[x]+|y_u-y_x|)\)
把绝对值拆掉,用两个树状数组维护就行了,分别对应前后缀查询

值得注意的是:
这题的重心要是带权重心,否则预处理就会 \(TLE\)

#include<bits/stdc++.h>
#define pb push_back
using namespace std;
const int N=3e5+10,M=22,inf=1e9;
int n,m=0,tt=0,cnt=0,b[N],s[N],Lx[N],Ly[N],fa[N];
vector<int>S[N],G[N],E[N];
map<int,int>id[N];
inline void priwork(){
    for(int i=1;i<=m;i++){
        if(S[i].empty())continue;
        sort(S[i].begin(),S[i].end());
        for(int j=0,si=S[i].size(),x,k=0;j<si;j++){
            x=S[i][j];id[i][x]=++tt;
            if(j && S[i][j-1]+1==x)G[tt].pb(tt-1),G[tt-1].pb(tt);
            else ++cnt,Lx[cnt]=i,Ly[cnt]=j;
            b[tt]=cnt;s[cnt]++;
            while(k<(int)S[i-1].size() && S[i-1][k]<x)k++;
            if(k<(int)S[i-1].size() && S[i-1][k]==x){
                E[b[id[i-1][x]]].pb(cnt),E[cnt].pb(b[id[i-1][x]]);
                G[id[i-1][x]].pb(tt),G[tt].pb(id[i-1][x]);
            }
        }
    }
    for(int i=1;i<=n;i++){
        sort(E[i].begin(),E[i].end());
        E[i].erase(unique(E[i].begin(),E[i].end()),E[i].end());
    }
}
int son[N]={inf},sz[N],rt=0,sum,p[N][M],dis[N][M];bool vis[N];
inline void getroot(int x,int last){
    sz[x]=s[x];son[x]=0;
    for(int i=E[x].size()-1;i>=0;i--){
        int u=E[x][i];
        if(u==last || vis[u])continue;
        getroot(u,x);sz[x]+=sz[u];son[x]=max(son[x],sz[u]);
    }
    son[x]=max(son[x],sum-sz[x]);
    if(son[x]<son[rt])rt=x;
}
vector<int>t1[N],t2[N];int dep[N];
inline void add1(int o,int x,int t){
    for(int i=x;i<=s[o];i+=(i&(-i)))t1[o][i]=min(t1[o][i],t);
}
inline int qry1(int o,int x){
    int ret=inf;
    for(int i=x;i>=1;i-=(i&(-i)))ret=min(ret,t1[o][i]);
    return ret;
}
inline void add2(int o,int x,int t){
    for(int i=x;i>=1;i-=(i&(-i)))t2[o][i]=min(t2[o][i],t);
}
inline int qry2(int o,int x){
    int ret=inf;
    for(int i=x;i<=s[o];i+=(i&(-i)))ret=min(ret,t2[o][i]);
    return ret;
}
inline void build(int x,int d){
    queue<int>Q;
    for(int i=Ly[x],u;i<Ly[x]+s[x];i++){
        u=id[Lx[x]][S[Lx[x]][i]];
        p[u][d]=i-Ly[x]+1;dis[u][d]=0;Q.push(u);
    }
    while(!Q.empty()){
        int x=Q.front();Q.pop();
        for(int i=G[x].size()-1;i>=0;i--){
            int u=G[x][i];
            if(vis[b[u]] || p[u][d])continue;
            p[u][d]=p[x][d],dis[u][d]=dis[x][d]+1,Q.push(u);
        }
    }
    for(int i=0;i<=s[x];i++)t1[x].pb(inf),t2[x].pb(inf);
}
inline void solve(int x,int d){
    vis[x]=1;dep[x]=d;build(x,d);
    for(int i=E[x].size()-1;i>=0;i--){
        int u=E[x][i];if(vis[u])continue;
        rt=0;sum=sz[u];getroot(u,x);
        fa[rt]=x;solve(rt,d+1);
    }
}
inline void Modify(int y){
    int x=b[y],d;
    while(x){
        d=dep[x];
        add1(x,p[y][d],dis[y][d]-p[y][d]);
        add2(x,p[y][d],dis[y][d]+p[y][d]);
        x=fa[x];
    }
}
inline int query(int y){
    int ret=inf,d,x=b[y];
    while(x){
        d=dep[x];
        ret=min(ret,dis[y][d]+p[y][d]+qry1(x,p[y][d]));
        ret=min(ret,dis[y][d]-p[y][d]+qry2(x,p[y][d]));
        x=fa[x];
    }
    if(ret<N)return ret;
    return -1;
}
int main(){
  int x,y,Q,op;
  scanf("%d",&n);
  for(int i=1;i<=n;i++){
      scanf("%d%d",&x,&y);
      m=max(x,m);S[x].pb(y);
  }
  priwork();
  rt=0;sum=n;getroot(1,1);solve(1,0);
  cin>>Q;
  while(Q--){
      scanf("%d%d%d",&op,&x,&y);
      if(op==2)printf("%d\n",query(id[x][y]));
      else Modify(id[x][y]);
  }
  return 0;
}

转载于:https://www.cnblogs.com/Yuzao/p/8711874.html