洛谷P2504[HAOI2006]聪明的猴子(kruskal)

洛谷P2504
题目大概意思是给n个树,都给出坐标,m个猴子,每个猴子的最大跳跃距离也给出,问有多少只猴子可以在这n棵树之间自由穿梭。

kruskal解法
计算出每两棵树之间的距离并建边,把所有边按照长度从小到大排序,遍历每条边,当连通块数量只剩一个的时候,就跳出,并且记录下访问的最后一条边的长度,用这个长度去和猴子的跳跃距离作比较即可。

#include<bits/stdc++.h>
using namespace std;
const int N=1e5+5;
const int mod=1e9+7;
const int INF=0x3f3f3f3f;
const double EPS=1e-6;
typedef long long ll;
#define IOS ios::sync_with_stdio(false);cin.tie(0);cout.tie(0);
int monkey[N],fa[N];
struct tree
{
    int x,y;
}t[N];
struct node
{
    int from,to;double dis;
}e[1000050];
int tot=0;
void addedge(int u,int v)
{
    tot++;
    e[tot].from=u;
    e[tot].to=v;
    e[tot].dis=sqrt((t[u].x-t[v].x)*(t[u].x-t[v].x)+(t[u].y-t[v].y)*(t[u].y-t[v].y));
}
bool cmp(node a,node b)
{
    return a.dis<b.dis;
}
int finds(int x)
{
    if(fa[x]==x)
        return x;
    return fa[x]=finds(fa[x]);
}
int main()
{
    IOS;
//    freopen("monkey.in","r",stdin);
//    freopen("monkey.out","w",stdout);
    int m;
    cin>>m;
    for(int i=1;i<=m;i++)
    {
        cin>>monkey[i];
    }
    int n;
    cin>>n;
    for(int i=1;i<=n;i++)
    {
        cin>>t[i].x>>t[i].y;
    }
    for(int i=1;i<=n;i++)
    {
        for(int j=1;j<=n;j++)
        {
            if(i!=j)
            addedge(i,j);
        }
    }
    sort(e+1,e+tot+1,cmp);
    for(int i=1;i<=n;i++)
    {
        fa[i]=i;
    }
    int cnt=n;
    double maxd=0;
    for(int i=1;i<=tot;i++)
    {
        if(cnt==1)
            break;
        int xx=finds(e[i].from);
        int yy=finds(e[i].to);
        if(xx!=yy)
        {
            fa[xx]=yy;
            cnt--;
            maxd=e[i].dis;
        }
    }
    int ans=0;
    for(int i=1;i<=m;i++)
    {
        if(maxd<=monkey[i])
        {
            ans++;
        }
    }
    cout<<ans;
}

  • 1
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 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、付费专栏及课程。

余额充值