POI2014 Hotel

最新推荐文章于 2020-04-07 10:19:09 发布

兔兔不会梦到ACM学长

最新推荐文章于 2020-04-07 10:19:09 发布

阅读量337

点赞数

分类专栏： POI题解文章标签： POI

本文链接：https://blog.csdn.net/WhiteAndGold/article/details/82889087

版权

POI题解专栏收录该内容

26 篇文章 2 订阅

订阅专栏

Hotel

POI2014

题意

1.一棵n个节点的树

2.选三个点，令这三个点两两之间距离相同

3.问有多少种选点的方案

解

1.可以枚举一个点为根
2.在根下面选三棵子树
3.每个子树里选一条从根出发且长度相等的链
4.定义
cnt[x][0]:长度为x的链有多少条
cnt[x][1]:长度为x的链有多少对

具体操作

1.清空cnt数组
2.枚举一个点为根
3.选一棵子树遍历
4.先把原本的答案拷贝下来
5.遍历时，用旧的答案更新得到新的答案

(滚动省空间)
cnt[1-cur]是之前合并完子树后的答案(旧答案)
cnt[cur]是当前合并完子树后的新答案(新答案)
ans+=cnt[1-cur][dis[x]][1];  
cnt[cur][dis[x]][1]+=cnt[1-cur][dis[x]][0];
cnt[cur][dis[x]][0]++;

具体代码

#include<bits/stdc++.h>
using namespace std;
const int M=5005;
int n,cnt[2][M][2],cur,dis[M];
long long ans;
int head[M],asdf;
struct edge {
    int to,nxt;
} G[M*2];
void add_edge(int a,int b) {
    G[++asdf].to=b;
    G[asdf].nxt=head[a];
    head[a]=asdf;
}
void dfs(int x,int f) {
    ans+=cnt[1-cur][dis[x]][1];
    cnt[cur][dis[x]][1]+=cnt[1-cur][dis[x]][0];
    cnt[cur][dis[x]][0]++;
    for(int i=head[x]; i; i=G[i].nxt) {
        int y=G[i].to;
        if(y==f)continue;
        dis[y]=dis[x]+1;
        dfs(y,x);
    }
}
int main() {
    int a,b;
    scanf("%d",&n);
    for(int i=1; i<n; i++) {
        scanf("%d %d",&a,&b);
        add_edge(a,b);
        add_edge(b,a);
    }
    for(int x=1; x<=n; x++) {
        memset(cnt,0,sizeof(cnt));
        cur=0;
        for(int i=head[x]; i; i=G[i].nxt) {
            cur=1-cur;
            memcpy(cnt[cur],cnt[1-cur],sizeof(cnt[cur]));
            int y=G[i].to;
            dis[y]=1;
            dfs(y,x);
        }
    }
    printf("%lld\n",ans);
    return 0;
}