见面礼——图论

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

Input

输入的第一行一个整数 n(2≤n≤105) 表示节点数,接下来 n 行每行两个整数 x,y(1≤x,y≤n) 描述图上的一条边。保证图中没有重边自环。

Output

输出一行一个正整数表示答案。

Input

6

1 2

1 3

1 4

1 5

1 6

2 3

Output

10

解析:

n个点n条边,所以该图就成一个环。只有将环中的一条边删去,该图才能变为一棵树。

复制代码
#include <bits/stdc++.h>
using namespace std;
#define int long long 
#define ios ios::sync_with_stdio(false),cin.tie(0),cout.tie(0);
typedef pair<int,int> PII;
const int N=2e6+10;
vector <int> g[N];
map <int,int> k;
int p[N];
int d[N];
bool vis[N];
vector <PII> q;  //储存成环的边
int n;
void dfs(int u,int pa)
{
    p[u]=pa;
    vis[u]=1;
    for (auto v:g[u])
    {
        if (v==pa) continue;
        if (vis[v]==1&&q.size()==0)    //当点 v 被再次遍历时,现在已经建成一个环了,就可以回溯将环中的每条边放入队列 q 中
        {
            q.push_back({u,v});
            while (p[u]!=v)
            {
                q.push_back({u,p[u]});
                u=p[u];
            }
            q.push_back({u,p[u]});
        }
        if (vis[v]==1) continue;  //走过的点,不用继续操作了,否则会死循环
        dfs(v,u);
    }
}
signed main()
{
    ios;
    cin>>n;
    for (int i=1;i<=n;i++)
    {
        int u,v;
        cin>>u>>v;
        g[u].push_back(v);
        g[v].push_back(u);
        d[u]++;
        d[v]++;
    }
    for (int i=1;i<=n;i++)
    {
        k[d[i]]++;  //记录度数相同的点的数量
    }
    //for (auto x:k) cout<<x.first<<" "<<x.second<<endl;
    dfs(1,0);
    //for (auto x:q) cout<<x.first<<" "<<x.second<<endl;
    int ans=0;
    for (auto x:q)   //遍历每条要删掉的边
    {
        int du=d[x.first];
        int dv=d[x.second];
        k[du]--;
        k[dv]--;
        k[du-1]++;
        k[dv-1]++;
        int res=0;
        bool flag=0;
        for (auto y:k)       //删除边后,再遍历每个点,判断能否成为根节点
        {
            int cnt=y.first;
            int s=y.second;
            if (cnt<=3) res +=s;
            if (cnt>=5&&s>0) flag=1;    //既当不了根节点,也当不了儿子节点
        }
        if (flag==0) ans +=res;
        k[du]++;  //还原
        k[dv]++;
        k[du-1]--;
        k[dv-1]--;
    }
    cout<<ans;
    return 0;
}
相关推荐
shehuiyuelaiyuehao3 分钟前
算法42,模拟算法,模拟z字形变换
算法
朝朝辞暮i6 分钟前
C++第一课
开发语言·c++·算法
橘子汽水16813 分钟前
Leetcode 208,207实现Trie前缀树,课程表
java·数据结构·算法·leetcode
hanlin0317 分钟前
刷题笔记:力扣第169题-多数元素
笔记·算法·leetcode
hfywmsj24 分钟前
广州餐饮铺位招租数据建模:基于多维评估体系的选址算法设计与实现
人工智能·算法·广州餐饮铺位招租
wabs66633 分钟前
关于二叉树【力扣116.填充每个节点的下一个右侧节点指针的思考】
数据结构·c++·算法·leetcode·二叉树·层序遍历
船厂电气自动化ai大模型35 分钟前
AI大模型与数学 第73课 秩的核心定义、矩阵秩的通用求解方法
开发语言·数据结构·机器学习
动词ing38 分钟前
【学习笔记】数据结构(哈希表基础+哈希集合+哈希映射+计数+查找)
数据结构·学习·哈希算法
Nil2081 小时前
leetcode 74搜索二维矩阵
算法·leetcode·矩阵
新知图书1 小时前
第 14 章 后训练算法GRPO详解与实战
人工智能·算法