F - 见面礼

见面礼 - CSG 1242 - Virtual Judge (vjudge.net)

题意：

给定一个 n 个点 n 条边的无向图，你需要求有多少种选择图上的一个点 p 和一条边(x,y) 的方案，使得删去(x,y) 后图变成一棵树，且这棵树以 p 为根时每个节点的儿子个数均不超过 3。

考虑计算能选的边数和能选的点数，然后他们两个相乘在加上没算上去选法就是答案。

首先能选的边一定是在环里的。如果点的连边数量小于等于3，那么这个点一定能选。

如果点的连边数量等于4并且这个点是环内的点，那么会多两种合法方案。如下图所示会多出1-2 选1的方案和1- 3 选1 的方案。这两个选法都是合法的

如果点的连边数量等于5，那么能选的边的数量就永远只有两个，就是这个点连接环的那两条边。这点看题目给出的样例应该可以理解。

还有一种最极限的情况，如下图所示，出现了两次连边数量等于5的点。这种情况能删的边数只有一个，就是3 - 4这条边。需要特判一下。

所以答案就是合法的边数*合法的点数加上连边==4的点的合法方案。

代码如下：
 

#include <bits/stdc++.h>
using namespace std;
#define int long long
const int N = 2e5 + 10;
vector<int> side[N];
int cnt[N];
vector<int> path;

int res;
bool is = false;
int huan[N];
void dfs(int u, int pre)
{
    if (is)
        return;
    for (int v : side[u])
    {
        if (v == pre)
            continue;
        cnt[v]++;
        path.push_back(v);
        if (cnt[v] == 2)
        {
            is = true;
            huan[path.back()] = 1;
            path.pop_back();
            while (path.back() != v)
            {
                huan[path.back()] = 1;
                path.pop_back();
                res++;
            }
            res++;
            return;
        }
        dfs(v, u);
        if (is)
            return;
        cnt[v]--;
        path.pop_back();
    }
}
void solve()
{
    int n;
    cin >> n;
    for (int i = 1; i <= n; i++)
    {
        int a, b;
        cin >> a >> b;
        side[a].push_back(b);
        side[b].push_back(a);
    }
    side[0].push_back(1);
    dfs(0, -1);
    int res1 = res;
    int res2 = 0;
    int q = 0;
    int oppp = 0;
    for (int i = 1; i <= n; i++)
    {
        if (side[i].size() <= 3)
            res2++;
        if (side[i].size() == 4)
        {
            if (huan[i] && oppp == 0)
            {
                q += 2;
            }
        }
        if (side[i].size() == 5)
        {
            res1 = 2;
            oppp++;
            q = 0;
            for (int v : side[i])
            {
                if (huan[v])
                {
                    if (side[v].size() == 4)
                    {
                        q++;
                    }
                }
            }
        }
    }
    if (oppp == 2)
    {
        cout << res2 << endl;
        return;
    }
    // cout << res1 << ' ' << res2 << ' ' << q << endl;
    cout << (res1 * res2) + q << endl;
}
signed main()
{
    ios::sync_with_stdio(false);
    cin.tie(0);
    cout.tie(0);

    int t;
    t = 1;
    while (t--)
    {
        solve(); // WASD
    }
}