洛谷 P3884 [JLOI2009]二叉树问题

 

---

不仅数据弱，而且范围弱的一道题，然而就算这样一开始也错了好几次，而且这道题不让下载测试点，不甘心看题解的蒟蒻写了一个小时才拿下

一开始粗心的看了一下题目大意，以为下标都是有序的，直接用线段树来套了，结果5后面连着8和9..，而且寻找LCA（Latest Common Ancestors，最近公共祖先）的时候也有点问题，没有判断是否在同一个深度，这样找到的不一定是LCA（这道题数据这么弱直接暴力跳就行了）。

WA代码 

#include <bits/stdc++.h>
using namespace std;
const int MAXN=101;
const int BMAXN=101<<1|1;

bitset<BMAXN>t;

int ans,width,depth,n,x,y,u1,u2;
inline void scan(){
    scanf("%d",&n);
    int u,v;
    for(int i=1;i<n;i++){
        scanf("%d %d",&u,&v);
        t[u]=true;
        t[v]=true;
    }
    scanf("%d %d",&x,&y);
}
inline int fa(int i){
    return i>>1;
}
inline int lc(int i){
    return i<<1;
}
inline int rc(int i){
    return i<<1|1;
}
void dfsDepth(int i,int level){
    if(!t[lc(i)]&&!t[rc(i)])depth=max(depth,level);
    if(t[lc(i)])dfsDepth(lc(i),level+1);
    if(t[rc(i)])dfsDepth(rc(i),level+1);
}
void dfsWidth(int l,int r){
    int sum=0;
    for(int i=l;i<=r;i++){
        sum+=t[i];
    }
    if(sum==0)return;
    width=max(width,sum);
    dfsWidth(lc(l), rc(r));
}
int nodeDepth(int i){
    int w=1,r=1;
    while(r<i){
        r=r<<1|1;
        w++;
    }
    return w;
}
void ufs(int l,int r){
    if(l==r)return;

    int ld= nodeDepth(l),rd= nodeDepth(r);
    if(ld>rd){
        u1++;
        ufs(fa(l),r);
    }else if(rd>ld){
        u2++;
        ufs(l, fa(r));
    }else{
        u1++;u2++;
        ufs(fa(l), fa(r));
    }
}

int main(){
    scan();

    dfsDepth(1,1);
    dfsWidth(1,1);
    ufs(x,y);

    cout<<depth<<endl;
    cout<<width<<endl;
    cout<<u1*2+u2;
    return 0;
}

因为要寻找祖先所以需要在输入的时候确定好父节点的下标，因为要判断深度所以在dfs的时候同时把深度确定下来，同时要输出宽度最大值，直接d1[d[i]]++嵌套深度找出同一深度的最大值。

一套军体拳下来比上面那个代码短了不少。

AC代码

#include <bits/stdc++.h>
using namespace std;
const int MAXN=100+1;

int n,x,y,depth,width,u1,u2;
pair<int,int>node[MAXN];
queue<int>q;
int f[MAXN],d[MAXN],d1[MAXN];
inline void scan(){
    cin>>n;
    int u,v;
    for(int i=1;i<n;i++){
        cin>>u>>v;
        if(!node[u].first)node[u].first=v;
        else node[u].second=v;
        f[v]=u;
    }
    cin>>x>>y;
}
void dfs(int i,int level){
    if(!node[i].first&&!node[i].second)depth=max(depth,level);
    d[i]=level;
    d1[d[i]]++;
    width=max(d1[d[i]],width);
    if(node[i].first)dfs(node[i].first,level+1);
    if(node[i].second)dfs(node[i].second,level+1);
}
void ufs(int l,int r){
    if(l==r)return;

    if(d[l]>d[r]){
        u1++;
        ufs(f[l],r);
    }else if(d[l]<d[r]){
        u2++;
        ufs(l,f[r]);
    }else{
        u1++;u2++;
        ufs(f[l],f[r]);
    }
}

int main(){
    scan();
    dfs(1,1);
    ufs(x,y);

    cout<<depth<<endl;
    cout<<width<<endl;
    cout<<u1*2+u2;
    return 0;
}

AC代码里面的queue<int>q是本蒟蒻打算用bfs记录最大深度值用的....