洛谷 P2341 [HAOI2006]受欢迎的牛

6 篇文章 0 订阅
3 篇文章 0 订阅

题目背景

Tarjan的模板题。

题目描述

每头奶牛都梦想成为牛棚里的明星。被所有奶牛喜欢的奶牛就是一头明星奶牛。所有奶

牛都是自恋狂,每头奶牛总是喜欢自己的。奶牛之间的“喜欢”是可以传递的——如果A喜

欢B,B喜欢C,那么A也喜欢C。牛栏里共有N 头奶牛,给定一些奶牛之间的爱慕关系,请你

算出有多少头奶牛可以当明星。

输入输出格式

输入格式:

第一行:两个用空格分开的整数:N和M

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

输出格式:

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

输入输出样例

输入样例:

3 3
1 2
2 1
2 3

输出样例:

1
说明

只有 3 号奶牛可以做明星

【数据范围】

10%的数据N<=20, M<=50

30%的数据N<=1000,M<=20000

70%的数据N<=5000,M<=50000

100%的数据N<=10000,M<=50000

题解

Tarjan裸题。

这道题就是Tarjan把环缩一下,然后统计出度为零的点,为1则输出答案,更大则这张图不是连通图。

代码

#include<bits/stdc++.h>
#define N 50050
using namespace std;
stack<int> s;
int n,m,x,y,top,scc_cnt,T,num_e,num_E;
int head[N],Head[N],low[N],dfn[N],num[N],scc[N],cd[N];
bool b[N];
struct Edge{
    int nex,to;
    Edge(){}
    Edge(int _,int __):nex(_),to(__){}
}e[N],E[N];
void add(int x,int y){
    e[++num_e]=Edge(head[x],y);
    head[x]=num_e;
}
void Add(int x,int y){
    E[++num_E]=Edge(Head[x],y);
    Head[x]=num_E;
}
void tarjan(int x){
    dfn[x]=low[x]=++T;s.push(x);
    b[x]=true;
    for(int i=head[x];i;i=e[i].nex){
        if(!dfn[e[i].to]){
            tarjan(e[i].to);
            low[x]=min(low[x],low[e[i].to]);
        }
        else if(b[e[i].to]) low[x]=min(low[x],dfn[e[i].to]);
    }
    if(low[x]==dfn[x]){
        int v;scc_cnt++;
        do{
            v=s.top();s.pop();
            scc[v]=scc_cnt;
            num[scc_cnt]++;
            b[v]=false;
        }while(v!=x);
    }
}
int main(){
    scanf("%d%d",&n,&m);
    for(int i=1;i<=m;++i){
        scanf("%d%d",&x,&y);
        add(x,y);
    }
    for(int i=1;i<=n;i++){
        if(!dfn[i]){
            top=0;
            tarjan(i);
        }
    }
    for(int i=1;i<=n;i++){
        for(int j=head[i];j;j=e[j].nex){
            if(scc[i]==scc[e[j].to]) continue;
            cd[scc[i]]++;
            Add(scc[i],scc[e[j].to]);
        }
    }
    int numm=0,ans=0;
    for(int i=1;i<=scc_cnt;i++){
        if(cd[i]==0) numm++,ans=i;
    }
    if(numm==1) printf("%d",num[ans]);
    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$ 位置上的值; - 在插入元素时,需要将离散化后的数列的元素从右往左依次插入树状数组中,而不是从左往右。
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值