洛谷 P2341 [HAOI2006]受欢迎的牛

该博客讨论了洛谷P2341题目的解决方案,涉及奶牛之间的喜欢关系。问题要求找出牛棚中可能成为明星奶牛的数量,即被所有奶牛喜欢的奶牛。通过 Tarjan 算法求解强连通分量并缩点,判断出度为0的点来确定明星奶牛的数量。博客提供了数据范围和样例,并给出了具体的代码实现。

题目描述

每头奶牛都梦想成为牛棚里的明星。被所有奶牛喜欢的奶牛就是一头明星奶牛。所有奶牛都是自恋狂,每头奶牛总是喜欢自己的。奶牛之间的“喜欢”是可以传递的——如果 A 喜欢 BB 喜欢 C,那么 A 也喜欢 C 。牛栏里共有 N 头奶牛,给定一些奶牛之间的爱慕关系,请你算出有多少头奶牛可以当明星。

输入输出格式

输入格式:

第一行:两个用空格分开的整数:NM
第二行到第 M+1 行:每行两个用空格分开的整数:AB,表示 A 喜欢 B

输出格式:

第一行:单独一个整数,表示明星奶牛的数量

输入输出样例

输入样例#1:
3 3
1 2
2 1
2 3
输出样例#1:
1

说明

只有 3 号奶牛可以做明星

数据范围

10% 的数据N20,M50

30% 的数据N1000,M20000

70% 的数据N5000,M50000

100% 的数据N10000,M50000


solution

  • 一道比较裸的缩点

  • 先跑一遍Tarjan求出强连通分量,然后缩点,统计每个点的出度

  • 如果只有一个点的出度为 0 ,那图大概是这样的

qwq

  • 这样 E 这个点的奶牛就被所有的奶牛喜欢啦,所以他们他们都是明星,直接输出 E 这个点包含多少奶牛

  • 如果不止一个点的出度为 0,那图大概是这样的

qwq

  • 这样没有奶牛被所有的奶牛喜欢,应该直接输出 0

code

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
typedef long long ll;

template<typename ____>
void _____(____ &___) {//听说这么写读入优化会增加代码有bug的几率qwq
    ___=0; ____ __=1;
    register char _=getchar();
    for(;_<'0'||_>'9';_=getchar())
        if(_=='-') __=-1;
    for(;_>='0'&&_<='9';_=getchar())
        ___=___*10+_-'0';
    ___*=__;
    return;
}

#define MAXN 10010
#define MAXM 50010

struct Edge {
    int u,v,next;
    Edge(int u=0,int v=0,int next=0):
        u(u),v(v),next(next) {}
};

Edge edge[MAXM];
int head[MAXN],cnt;

void addedge(int u,int v) {
    edge[++cnt]=Edge(u,v,head[u]);
    head[u]=cnt;
    return;
}

int dfn[MAXN],low[MAXN],timee;
int st[MAXN],top;
int col[MAXN],numcolor;
bool vis[MAXN],ins[MAXN];
int num[MAXN];

void dfs(int now) {
    dfn[now]=low[now]=++timee;
    st[++top]=now;
    vis[now]=ins[now]=true;
    for(int i=head[now];i;i=edge[i].next) {
        int v=edge[i].v;
        if(ins[v]) low[now]=min(low[now],dfn[v]);
        else if(!vis[v]) {
            dfs(v);
            low[now]=min(low[now],low[v]);
        }
    }
    if(dfn[now]==low[now]) {
        col[now]=++numcolor;
        num[numcolor]++;
        while(st[top]!=now) {
            col[st[top]]=numcolor;
            num[numcolor]++;
            ins[st[top--]]=false;
        }
        ins[now]=false;
        top--;
    }
    return;
}


int outdeg[MAXN];

int main() {
    int n,m;
    _____(n),_____(m);
    for(int i=1,A,B;i<=m;i++) {
        _____(A),_____(B);
        addedge(A,B);
    }
    for(int i=1;i<=n;i++)
        if(!vis[i]) dfs(i);
    for(int u=1;u<=n;u++)
        for(int i=head[u];i;i=edge[i].next)
            if(col[u]!=col[edge[i].v])
                outdeg[col[u]]++;
    int sum=0,color;
    for(int i=1;i<=numcolor;i++)
        if(outdeg[i]==0) sum++,color=i;
    if(sum==1) printf("%d\n",num[color]);
    else printf("0");
    return 0;
}
这道题目还可以使用树状数组或线段树来实现,时间复杂度也为 $\mathcal{O}(n\log n)$。这里给出使用树状数组的实现代码。 解题思路: 1. 读入数据; 2. 将原数列离散化,得到一个新的数列 b; 3. 从右往左依次将 b 数列中的元素插入到树状数组中,并计算逆序对数; 4. 输出逆序对数。 代码实现: ```c++ #include <cstdio> #include <cstdlib> #include <algorithm> const int MAXN = 500005; struct Node { int val, id; bool operator<(const Node& other) const { return val < other.val; } } nodes[MAXN]; int n, a[MAXN], b[MAXN], c[MAXN]; long long ans; inline int lowbit(int x) { return x & (-x); } void update(int x, int val) { for (int i = x; i <= n; i += lowbit(i)) { c[i] += val; } } int query(int x) { int res = 0; for (int i = x; i > 0; i -= lowbit(i)) { res += c[i]; } return res; } int main() { scanf("%d", &n); for (int i = 1; i <= n; ++i) { scanf("%d", &a[i]); nodes[i] = {a[i], i}; } std::sort(nodes + 1, nodes + n + 1); int cnt = 0; for (int i = 1; i <= n; ++i) { if (i == 1 || nodes[i].val != nodes[i - 1].val) { ++cnt; } b[nodes[i].id] = cnt; } for (int i = n; i >= 1; --i) { ans += query(b[i] - 1); update(b[i], 1); } printf("%lld\n", ans); return 0; } ``` 注意事项: - 在对原数列进行离散化时,需要记录每个元素在原数列中的位置,便于后面计算逆序对数; - 设树状数组的大小为 $n$,则树状数组中的下标从 $1$ 到 $n$,而不是从 $0$ 到 $n-1$; - 在计算逆序对数时,需要查询离散化后的数列中比当前元素小的元素个数,即查询 $b_i-1$ 位置上的值; - 在插入元素时,需要将离散化后的数列的元素从右往左依次插入树状数组中,而不是从左往右。
评论
成就一亿技术人!
拼手气红包6.0元
还能输入1000个字符
 
红包 添加红包
表情包 插入表情
 条评论被折叠 查看
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值