提出 #73929150

---

ソースコード 拡げる

/*Sin_qwq's code
We can be heros , just for one day .*/
#include <bits/stdc++.h>
#define LD long double
#define i7 __int128
#define re return
#define con continue
#define ci const int
#define all(x) x.begin(), x.end()
#define All(x) x.begin() + 1, x.end()
using namespace std;
typedef long long i64;
template <class T>
inline bool ckmin(T &a, T b) { re b < a ? a = b, 1 : 0; }
template <class T>
inline bool ckmax(T &a, T b) { re a < b ? a = b, 1 : 0; }

vector<string> ans;
vector<int> a;
map<int, int> cnt;
vector<vector<int>> adj;
bool has_dup; // 标记当前路径是否已有重复

// 新增：回溯时恢复has_dup的状态
void dfs(int u, int parent)
{
    bool cur_dup = false;        // 记录当前节点是否导致重复
    bool prev_has_dup = has_dup; // 保存进入节点前的has_dup状态

    // 第一步：判断当前节点的答案
    if (has_dup)
    {
        // 路径已有重复，直接输出Yes
        ans[u] = "Yes";
    }
    else
    {
        if (cnt[a[u]] >= 1)
        {
            // 当前节点数值重复，标记路径有重复
            ans[u] = "Yes";
            has_dup = true;
            cur_dup = true;
        }
        else
        {
            // 无重复
            ans[u] = "No";
        }
    }

    // 第二步：将当前节点加入计数
    cnt[a[u]]++;

    // 第三步：遍历子节点
    for (auto v : adj[u])
    {
        if (v != parent)
        {
            dfs(v, u);
        }
    }

    // 第四步：回溯
    cnt[a[u]]--;
    if (cnt[a[u]] == 0)
    {
        cnt.erase(a[u]);
    }
    // 恢复has_dup：只有当前节点导致的重复，才恢复
    if (cur_dup)
    {
        has_dup = prev_has_dup;
    }
}

void work()
{
    int n;
    cin >> n;
    ans.resize(n + 1);
    a.resize(n + 1);
    adj.resize(n + 1);
    has_dup = false; // 初始路径无重复

    for (int i = 1; i <= n; i++)
    {
        cin >> a[i];
    }

    for (int i = 1; i <= n - 1; i++)
    {
        int u, v;
        cin >> u >> v;
        adj[u].push_back(v);
        adj[v].push_back(u);
    }

    dfs(1, -1);

    for (int i = 1; i <= n; i++)
    {
        cout << ans[i] << '\n';
    }
}

int main()
{
    ios::sync_with_stdio(false);
    cin.tie(nullptr);
    cout.tie(nullptr); // 加速输出
    int T = 1;
    // cin>>T;
    while (T--)
    {
        work();
    }
    re 0;
}

提出情報

ジャッジ結果