美丽的路径 并查集,搜索,二分

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

链接https://ac.nowcoder.com/acm/contest/9985/A

知识点

并查集,搜索,二分

思路
k 2 + 1 \frac{k}{2}+1 2k+1小的数,那就是中位数,需要注意的是题目中没有说要是最短路,所以为了让美丽值最大,可以让这条路径在两个权值大的点之间反复行走。

那么可以发现会有很多个美丽值,那么可以对答案进行二分,去找最大的那个。

首先是判断起点到终点之间是否联通,这个用搜索,或者并查集都可以。

当可以连通的时候,二分。设当前美丽值为x,路径的美丽值就是此路径上的所有点的点权中第⌊k/2+1⌋小的点权,等价于大于等于x的点的数量大于等于路径中所有点的一半。

换句话说,如果发现大于等于x的点的数量大于等于路径中所有点的一半,说明此时的x值取小了, l = m + 1 l=m+1 l=m+1,否则 r = m − 1 r=m-1 r=m1.

如何判断大于等于x的点的数量大于等于路径中所有点数量的一半?

设大于等于x的点为1点,否则为0点。

当有两个1点连续时,可以不断来回走,必然大于一半。

只有当10101010交错出现时,且1的个数大于0的个数,必然大于一半。

容易发现只有0101010交错出现时,且开始点和结束点都为0,才会小于。

代码

#include <bits/stdc++.h>
#include <iostream>
using namespace std;
const int maxn = 200005;
typedef long long ll;

int a[maxn];
int vis[maxn];
int fa[maxn];

vector<int> g[maxn];
int T,n,m,s,t;
int ans =0;
bool flag = false;

int find(int x){
    return fa[x] == x ? x : fa[x] = find(fa[x]); 
}
void dfs(int u,int x)
{
    vis[u] = 1;
    for(auto v : g[u]){
        if(a[u] >= x && a[v] >= x)
            flag = true;
        if(!vis[v])
            dfs(v,x);
    }
}

void dfs2(int u,int x)
{
    vis[u] = 1;
    for(auto v : g[u]){
        if(!vis[v] && (a[u] >= x) != (a[v] >= x))
            dfs2(v,x);
    }
}

int check(int x){
    memset(vis,0,sizeof(vis));
    flag = false;
    dfs(s,x);
    if(flag)
        return 1;
    if(a[s] < x && a[t] < x)
        return false;
    memset(vis,0,sizeof(vis));
    dfs2(s,x);
    if(vis[t] == 1)
        return 1;
    else
        return 0;
}

int main()
{
    cin.tie(0);
    ios::sync_with_stdio(false);
    cin>>T;
    while(T--){
        cin>>n>>m>>s>>t;
        for(int i = 1;i <= n;i++){
            g[i].clear();
            fa[i] = i;
            cin>>a[i];
        }
        int u,v;
        for(int i = 1;i <= m;i++){
            cin>>u>>v;
            g[u].push_back(v);
            g[v].push_back(u);
            fa[find(u)] = find(v);
        }
        if(find(s) == find(t)){
            cout<<"YES"<<endl;
            int l = 1,r = 1e9;
            while( l <= r){
                int m = (l + r) >> 1;
                if(check(m))
                    l = m + 1;
                else
                    r = m - 1;
            }
            cout<<r<<endl;
        }
        else
            cout<<"NO"<<endl;
    }
    return 0;
}
  • 1
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值