7-98 病毒溯源 (25 分) 树的dfs + 并查集

一、题目

PTA病毒溯源

二、分析

根据病毒的变异方向进行建树，对树进行深度优先搜索，搜到叶节点对比保存路径。

在建树的同时维护一个并查集，通过并查集找到树的根节点（病毒的源头）。

三、代码

#include <bits/stdc++.h>
using namespace std;
const int N = 10010;

int n, m;
int h[N], e[N * 2], ne[N * 2], idx;
bool st[N];
vector<int> v, s;
int f[N];

void add(int a, int b)
{
   e[idx] = b;
   ne[idx] = h[a];
   h[a] = idx++;
}

int find(int x)
{
   if (f[x] != x)
      f[x] = find(f[x]);
   return f[x];
}

vector<int> fmax(vector<int> a, vector<int> b)
{
   if (a.size() != b.size())
   {
      if (a.size() > b.size())
         return a;
      return b;
   }
   for (int i = 0; i < a.size(); i++)
   {
      if (a[i] > b[i])
         return b;
      if (b[i] > a[i])
         return a;
   }
   return a;
}

void dfs(int root)
{
   int p = 0;
   for (int i = h[root]; ~i; i = ne[i])
   {
      int j = e[i];
      if (!st[j])
      {
         p++;
         s.push_back(j);
         st[j] = true;
         dfs(j);
         st[j] = false;
         s.erase(s.end() - 1);
      }
   }
   if (!p && s.size() >= m)
   {
      m = s.size();
      v = fmax(s, v);
   }
}

int main()
{
   memset(h, -1, sizeof h);
   cin >> n;
   for (int i = 0; i < n; i++)
      f[i] = i;
   for (int i = 0; i < n; i++)
   {
      int k;
      cin >> k;
      while (k--)
      {
         int x;
         cin >> x;
         add(i, x), add(x, i);
         f[find(x)] = find(i);
      }
   }

   int root = find(0);
   s.push_back(root);
   st[root] = true;
   dfs(root);

   cout << m << endl;
   for (int i = 0; i < v.size(); i++)
   {
      i ? cout << " " << v[i] : cout << v[i];
   }

   return 0;
}