BZOJ3700: 发展城市

BZOJ3700: 发展城市

https://lydsy.com/JudgeOnline/problem.php?id=3700

分析:

• 枚举两个人，先求链交，求到两个端点的时间。
• 链交求法：求两两\(lca\)的\(4\)个点，去除掉不在路径上的，再去重，取\(dfs\)序最大的两个点。
• 同向，早进去的那个碗出来，异向，时间段有交。

代码:

#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
#define N 100050
typedef long long ll;
typedef double f2;
int head[N],nxt[N<<1],to[N<<1],cnt,val[N<<1],n;
int m,qs[N],qt[N],bg[N],ed[N],et[N<<1],qc[N],dep[N];
int f[20][N<<1],Lg[N<<1],dis[N],dfn[N];
char buf[100000],*p1,*p2;
#define nc() (p1==p2&&(p2=(p1=buf)+fread(buf,1,100000,stdin),p1==p2)?EOF:*p1++)
int rd() {
    int x=0; char s=nc();
    while(s<'0'||s>'9') s=nc();
    while(s>='0'&&s<='9') x=(((x<<2)+x)<<1)+s-'0',s=nc();
    return x;
}
int qv[N];
int ans;
inline void add(int u,int v,int w) {
    to[++cnt]=v; nxt[cnt]=head[u]; head[u]=cnt; val[cnt]=w;
}
bool in_subtree(int x,int y) {
    return dfn[y]>=dfn[x]&&dfn[y]<=ed[x];
}
int cmp(const int &x,const int &y) {return dep[x]<dep[y]?x:y;}
int lca(int x,int y) {
    x=bg[x]; y=bg[y];
    if(x>y) swap(x,y);int len=Lg[y-x+1];
    return cmp(f[len][x],f[len][y-(1<<len)+1]);
}
void dfs(int x,int y) {
    dfn[x]=++dfn[0];
    dep[x]=dep[y]+1;
    et[++et[0]]=x;
    int i;
    for(i=head[x];i;i=nxt[i]) if(to[i]!=y) {
        dis[to[i]]=dis[x]+val[i];
        dfs(to[i],x); et[++et[0]]=x;
    }
    ed[x]=dfn[0];
}
int Dis(int x,int y) {return dis[x]+dis[y]-2*dis[lca(x,y)];}
inline bool cmp1(const int &x,const int &y) {return dfn[x]>dfn[y];}
int G[5];
void QuErY(int x,int y) {
    G[1]=lca(qs[x],qs[y]);
    G[2]=lca(qs[x],qt[y]);
    G[3]=lca(qt[x],qs[y]);
    G[4]=lca(qt[x],qt[y]);
    int i;
    for(i=1;i<=4;i++) if(!in_subtree(qc[x],G[i])||!in_subtree(qc[y],G[i])) G[i]=0;
    sort(G+1,G+5,cmp1);
    int ln=unique(G+1,G+5)-G-1;
    for(i=1;i<=ln;i++) if(!G[i]) ln=i-1;
    if(!ln) return ;
    if(ln==1) {
        int u=G[1];
        ll d1=Dis(qs[x],u),d2=Dis(qs[y],u);
        if(d1*qv[y]==d2*qv[x]) ans++;
        return ;
    }else {
        int u=G[1],v=G[2];
        ll d1=Dis(qs[x],u),d2=Dis(qs[x],v),d3=Dis(qs[y],u),d4=Dis(qs[y],v);
        if(d1<=d2) {
            if(d3<=d4) {
                if((d1*qv[y] >= d3*qv[x] && d2*qv[y] <= d4*qv[x])||(d1*qv[y] <= d3*qv[x] && d2*qv[y] >= d4*qv[x])) ans++;
            }else {
                swap(d3,d4);
                if((d2*qv[y]>=d3*qv[x]&&d1*qv[y]<=d4*qv[x])||(d4*qv[x]>=d1*qv[y]&&d3*qv[x]<=d2*qv[y])) ans++;
            }
        }else {
            swap(d1,d2);
            if(d3>d4) {
                swap(d3,d4);
                if((d1*qv[y] >= d3*qv[x] && d2*qv[y] <= d4*qv[x])||(d1*qv[y] <= d3*qv[x] && d2*qv[y] >= d4*qv[x])) ans++;
            }else {
                if((d2*qv[y]>=d3*qv[x]&&d1*qv[y]<=d4*qv[x])||(d4*qv[x]>=d1*qv[y]&&d3*qv[x]<=d2*qv[y])) ans++;
            }   
        }
    }
}
void clr() {
     cnt=et[0]=0; dfn[0]=0; ans=0;
}
void solve() {
    n=rd();
    int i,x,y,z,j;
    clr();
    for(i=1;i<=n;i++) head[i]=0;
    for(i=1;i<n;i++) {
        x=rd(); y=rd(); z=rd();
        add(x,y,z); add(y,x,z);
    }
    dfs(1,0);
    for(i=et[0];i;i--) bg[et[i]]=i;
    for(i=1;i<=et[0];i++) f[0][i]=et[i];
    for(Lg[0]=-1,i=1;i<=et[0];i++) Lg[i]=Lg[i>>1]+1;
    for(i=1;(1<<i)<=et[0];i++) {
        for(j=1;j+(1<<i)-1<=et[0];j++) {
            f[i][j]=cmp(f[i-1][j],f[i-1][j+(1<<(i-1))]);
        }
    }
    dfn[0]=0;
    m=rd();
    for(i=1;i<=m;i++) qs[i]=rd(),qt[i]=rd(),qv[i]=rd(),qc[i]=lca(qs[i],qt[i]);
    for(i=1;i<=m;i++) {
        for(j=i+1;j<=m;j++) {
            QuErY(i,j);
        }
    }
    printf("%d\n",ans);
}
int main() {
    int Tcas;
    scanf("%d",&Tcas);
    while(Tcas--) solve();
}

转载于:https://www.cnblogs.com/suika/p/10092522.html