美丽的路径(二分中位数+check by dfs)

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

有n个点，m条边，每个点有一个全值。起点是s，终点是t，设美丽路径为s到t路程中第k/2+1小的数，求最大的美丽值是多少。

---

思路:

第k/2+1小的数，sort后其实就是中位数把。关于中位数的题一般是二分的trick。过一段时间我把二分中位数，二分平均值的题再补一下。

然后题目的情况如果有两个连着的1，因为可以来回一直走这两个1，所以最后结果>mid-->l=mid;

然后另外一种情况就是要使1的个数大于等于0，起点和终点不能都是0。(都0的话总数/2+1永远是<)

所以我们第二遍从起点出发沿着101010…(或者01）

这样找下去，如果可以到终点，那么就是可以的

关于找路径，dfs就好辣

---

知识点补充:

寻找中位数有一个通常的套路：
考虑二分中位数，设x 是现在二分的数,mid是序列的中位数：
将大于x 的数设为1 ，小于x 的数设为−1。[或者设1，0]
将整个1/−1序列求和
若Sum>0，说明1的数量比−1多，也就是大于x的数比小于x的数多
说明x<=mid
反之x>mid

若将题目中的中位数换成算数平均值.

check答案的时候，将序列每个数同时减去答案x.然后找到一个区间累和大于等于0的即合法。

因为将每个数同时减去答案x后，算数平均值≥0的即原序列算数平均值≥x.

---

代码:

#include<iostream>
#include<vector>
#include<queue>
#include<cstring>
#include<cmath>
#include<map>
#include<set>
#include<cstdio>
#include<algorithm>
#define debug(a) cout<<#a<<"="<<a<<endl;
using namespace std;
const int maxn=2e5+1000;
typedef long long LL;
inline LL read(){LL x=0,f=1;char ch=getchar();	while (!isdigit(ch)){if (ch=='-') f=-1;ch=getchar();}while (isdigit(ch)){x=x*10+ch-48;ch=getchar();}
return x*f;}
vector<LL>g[maxn];
LL val[maxn],n,m,s,t;
bool vis[maxn],flag=1;
void predfs(LL u){vis[u]=true;for(LL i=0;i<g[u].size();i++){LL v=g[u][i];if(vis[v]) continue;predfs(v);}}
void dfs1(LL u,LL mid){
    vis[u]=1;
    for(LL i=0;i<g[u].size();i++){
        LL v=g[u][i];
        if(val[u]>=mid&&val[v]>=mid) flag=0;
        if(vis[v]) continue;
        dfs1(v,mid);
    }
}
void dfs2(LL u,LL mid){
    vis[u]=1;
    for(LL i=0;i<g[u].size();i++){
        LL v=g[u][i];
        if(vis[v]) continue;
        if( (val[u]>=mid )!=(val[v]>=mid) )dfs2(v,mid);
    }
}
bool check(LL mid){
    memset(vis,0,sizeof(vis));
    flag=1;
    dfs1(s,mid);
    if(0==flag) return true;
    if(val[s]<mid&&val[t]<mid) return false;
    memset(vis,0,sizeof(vis));
    dfs2(s,mid);
    if(vis[t]) return true;
    else return false;
}
int main(void)
{
  cin.tie(0);std::ios::sync_with_stdio(false);
  LL T;cin>>T;
  while(T--){
    cin>>n>>m>>s>>t;
    for(LL i=1;i<=n;i++) cin>>val[i],g[i].clear(),vis[i]=0;
    for(LL i=1;i<=m;i++){
        LL u,v;cin>>u>>v;
        g[u].push_back(v);g[v].push_back(u);
    }
    predfs(s);
    if(!vis[t]){
        cout<<"NO"<<"\n";
        continue;
    }
    cout<<"YES"<<"\n";
    LL l=1;LL r=1e18;
    while(l<r){
        LL mid=(l+r+1)>>1;
        if(check(mid)) l=mid;
        else r=mid-1;
    }
    cout<<l<<"\n";
  }
return 0;
}

---

后记:

https://www.luogu.com.cn/problem/P2839（middle）

https://www.luogu.com.cn/problem/CF484E

CF1486D - Max Median

前面两道是二分中位数+主席树。后面这道是上次的D