【每日一题】Treepath

Treepath

http://www.nowcoder.com/questionTerminal/660faba350d74df095191ec3d321c15c

Solution

题目讲的比较明白,目的明确。就是每条边边权为1,求长为偶数的路径数。
我们知道 奇数+奇数=偶数;奇数+偶数=奇数;偶数+偶数=偶数;与异或运算比较相似。
我们用 图片说明 记录以i为根节点的0代表偶数边数,1代表奇数边数。
那么我们知道,叶子节点的偶数路径有一条0,奇数路径没有。图片说明
所以我们可以推出状态转移方程 图片说明
因为存在一条连向父节点的边,所以子节点只可以选择奇偶性不同的路径数相加。
得到上面递推式,我们在考虑更新答案,对于每个待处理父节点,已得到的偶数边*未处理过的子节点奇数边,或者奇偶互换,最终求和即得到最终偶数边个数。

时间复杂度 O(N)

Code:

#include <bits/stdc++.h>
using namespace std;
#define js ios::sync_with_stdio(false);cin.tie(0); cout.tie(0)
typedef long long ll;

inline int read() {
    int s = 0, w = 1; char ch = getchar();
    while (ch < 48 || ch > 57) { if (ch == '-') w = -1; ch = getchar(); }
    while (ch >= 48 && ch <= 57) s = (s << 1) + (s << 3) + (ch ^ 48), ch = getchar();
    return s * w;
}

const int N = 1e5 + 7;
vector<int>    e[N];
ll dp[N][2]; //0代表以i为根,偶数长度的边数,1代表奇数长度的边数
int n;
ll ans;

void dfs(int x, int fa) {
    dp[x][0] = 1; //初始化长度0为偶数的边有一条
    for (auto it : e[x]) {
        if (it == fa)    continue;
        dfs(it, x);
        ans += dp[x][0] * dp[it][1]; //以父节点被计算过的偶数边*未计算过的子节点奇数边(考虑连接父节点一条边)和为偶数,更新ans
        ans += dp[x][1] * dp[it][0]; //以父节点被计算过的奇数边*未计算过的子节点偶数边(考虑连接父节点一条边)和为偶数,更新ans
        dp[x][0] += dp[it][1];  //子节点奇数边+连接父节点边=偶数边合计父节点中
        dp[x][1] += dp[it][0];  //子节点偶数边+连接父节点边=奇数边合计父节点中
    }
}

int main() {
    n = read();
    for (int i = 1; i < n; ++i) {
        int u = read(), v = read();
        e[u].push_back(v); //建树
        e[v].push_back(u); 
    }
    dfs(1, 0);
    printf("%lld\n", ans);
    return 0;
}
每日一题 文章被收录于专栏

每日一题

全部评论

相关推荐

感性的干饭人在线蹲牛友:🐮 应该是在嘉定这边叭,禾赛大楼挺好看的
点赞 评论 收藏
分享
我即大橘:耐泡王
点赞 评论 收藏
分享
1 收藏 评论
分享
牛客网
牛客企业服务