微信公众号搜"智元新知"关注
微信扫一扫可直接关注哦!

带字符的树重建

如何解决带字符的树重建

给定一棵树,由 N 个顶点组成,以顶点 1 为根。每个顶点由一个小写英文字母作为节点值组成。

如果祖先节点与任何顶点的节点值相同,则该祖先节点被视为其有效父顶点。

我们的任务是从树中分离所有节点并将它们附加到任何有效的父顶点(在初始树中)并打印每个节点的子树中节点的计数。

这个问题是这个问题的延伸:https://www.hackerearth.com/practice/algorithms/graphs/depth-first-search/practice-problems/algorithm/tree-recreation-88871e63/

约束:

1

1

a

我的方法

使用堆栈,我首先找到重新创建树后形成的连接组件的数量。我想进一步使用我的代码来计算每个节点的子树数。任何人都可以分享他的方法吗?

  My code:
  
  #include<bits/stdc++.h>
 using namespace std;
  typedef long long ll;
typedef unsigned long long ull;
 #define M 1000000007
 #define ff first.first
 #define fs first.second
 #define sf second.first
 #define ss second.second
 #define f first
 #define s second
 #define inf 9000000000000000000
 vector<ll> adj[100002];
ll val[100002],compo;
stack<ll> st[26];
 ll cnt[26];
 ll arr[100002];

void dfs(ll x,ll par)
{
 if(st[val[x]].empty())
{
//     cout << x << " ";
    compo++;
}
st[val[x]].push(x);

for(auto node: adj[x])
{
    if(node==par)
    {
        continue;
    }      
    dfs(node,x);   

}
    
st[val[x]].pop();
}

int main()
{
ios_base::sync_with_stdio(false);
cin.tie(NULL); 
cout.tie(NULL);
ll t;
cin>>t;
while(t--)
{
    memset(cnt,sizeof(cnt));
    memset(arr,sizeof(arr));
    ll n,i;
    char s;
    compo = 0;
    cin>>n;
    for(i=1; i<=n; i++)
        adj[i].clear();

    for(i=1; i<=n; i++)
    {
        cin>>s;
        val[i] = s - 'a';
    }
    for(i=1; i<n; i++)
    {
        ll a,b;
        cin>>a>>b;
        adj[a].push_back(b);
        adj[b].push_back(a);
    }

    dfs(1,0);
    cout<<compo<<"\n";
   
    
}

return 0;
}

还要检查我的这个实现:https://pastebin.com/XTPQ1pfJ

版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。