圆桌就坐问题-算法题

题意:现有一张圆桌,有若干喜欢关系,每个人只在喜欢的人身旁就坐,问该圆桌最多能坐多少人;
共有n个人,每个人都只喜欢一个人(不能是自己)
传送门

题解:由于喜欢是单向的,所以可以通过构造有向图,根据有向图判断最大环即可,但有个问题,当两个人互相喜欢时,则两人的两侧可连接指向他们的最长边,因此需要特判互相喜欢的情况。

int maximumInvitations(vector<int>& favorite) {
    int n = favorite.size();
    vector<int> indeg(n), vis(n), dp(n, 1);
    for (auto &x : favorite) ++indeg[x];
    queue<int> q;
    for (int i = 0; i < n; ++i)
        if (!indeg[i]) 
        	q.push(i);

    while (!q.empty()) {
        int u = q.front();
        vis[u] = true;
        q.pop();
        int v = favorite[u];
        dp[v] = max(dp[v], dp[u]+1);
        --indeg[v];
        if (!indeg[v])
            q.push(v);
    }

    int ring = 0, total = 0;
    for (int i = 0; i < n; ++i) {
        if (!vis[i]) {
            int j = favorite[i];
            // 互相喜欢,累加答案
            if (favorite[j] == i) {
                total += dp[i] + dp[j];
                vis[i] = vis[j] = true;
            }
            else {
            	// 找最大环
                int u = i, cnt = 0;
                while (1) {
                    ++cnt;
                    u = favorite[u];
                    vis[u] = true;
                    if (u == i)
                        break;
                }
                ring = max(ring, cnt);
            }
        }
    }
    // 答案只能是 【互相喜欢+最长边】 或 【最大环】 二者之一
    return max(ring, total);
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值