P5836 [USACO19DEC] Milk Visits S

原題連結

題解

樹上只有兩種顏色，我們把每種顏色的連通塊記錄下來，只有當路徑兩端的點屬於同一連通塊且顏色與朋友喜歡的不同時輸出0

code

#include<bits/stdc++.h>
using namespace std;

char s[100005];
int fa[100005];
int finds(int now){return fa[now]==now?now:fa[now]=finds(fa[now]);}

vector<int> G[100005];

void dfs(int now,int pre)
{
    for(auto next:G[now])
    {
        if(pre==next) continue;
        dfs(next,now);
        if(s[next]==s[now]) fa[finds(next)]=now;
    }
}
int main()
{
    int n,m;
    cin>>n>>m;
    for(int i=1;i<=n;i++) fa[i]=i;
    cin>>(s+1);
    for(int i=1;i<n;i++)
    {
        int x,y;
        cin>>x>>y;
        G[x].push_back(y);
        G[y].push_back(x);
    }

    dfs(1,1);

    string ans;
    for(int i=1;i<=m;i++)
    {
        int x,y;
        cin>>x>>y;
        char a;
        cin>>a;
        if(finds(x)==finds(y)&&s[finds(x)]!=a) ans+='0';
        else ans+='1';
    }
    cout<<ans;
    return 0;
}