CodeForces_Tree Cutting (dfs)

最新推荐文章于 2024-03-24 17:02:33 发布

原创最新推荐文章于 2024-03-24 17:02:33 发布 · 159 阅读

0 ·

CC 4.0 BY-SA版权

ACM_图论、搜索专栏收录该内容

13 篇文章

订阅专栏

本文提供了一道Codeforces竞赛中F1题目的详细解答，采用深度优先搜索(DFS)和动态规划(DP)策略来解决一个关于树形结构的问题。代码展示了如何遍历树并计算特定条件下的路径数量。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目链接：https://codeforces.com/contest/1118/problem/F1

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int maxn=3e5+2;
int a[maxn];
int dp[maxn][3];
int sum1,sum2,n;
vector<int> v[maxn];
int ans=0;
void dfs(int t,int f){
    for(int i=0;i<v[t].size();i++){
        int u=v[t][i];
        if(u==f) continue;
        dfs(u,t);
        if(dp[u][1]==sum1&&dp[u][2]==0) ans++;
        else if(dp[u][1]==0&&dp[u][2]==sum2) ans++;      
        dp[t][1]+=dp[u][1];
        dp[t][2]+=dp[u][2];
    }
    dp[t][a[t]]++;
}
int main()
{
    cin>>n;
    for(int i=1;i<=n;i++){
        cin>>a[i];
        if(a[i]==1) sum1++;
        if(a[i]==2) sum2++;
    }
    for(int i=1;i<n;i++){
        int x,y;
        cin>>x>>y;
        v[x].push_back(y);
        v[y].push_back(x);
    }
    dfs(1,0);
    cout<<ans<<endl;
    return 0;
}