标签:include return vis int mid 牛客 2021 集训营 true
题目大意
有n个点,m条边,每个点有一个权值。起点是s,终点是t,设美丽路径为s到t路程中第k/2+1小的数,求最大的美丽值是多少。
解题思路
二分去取答案。我们把大于等于mid的权值设为1,小于mid为0。因为我们要求k/2+1小的数,即路径中第k/2+1小的数是1,1的个数要大于等于0的个数。
-
用并查集维护起点和终点,直接判-1的情况
-
如果有连着的两个1,那么最后结果一定是大于mid,因为可以来回一直走这两个1
-
去掉上述情况之后要使1的个数大于等于0,有两种情况:
- 点数为奇数,则要满足类似
10101...1
的结构。 - 点数为偶数,满足
1010..10
或0101..01
即可,即 0 0 0 和 1 1 1 交替出现。
实际实现的时候,只需要满足以下条件:
- 不论是奇数还是偶数,都满足路径上相邻的两点标记不同。
- 若为奇数,最后的标记是 1 1 1。
- 点数为奇数,则要满足类似
注意事项
1.二分的左端点为读入的最小值,右端点为最大值,可以使程序跑的快一点。
参考代码
#include<stdio.h>
#include<iostream>
#include<vector>
#include<cstring>
#include<cstdio>
#include<climits>
#include<cmath>
#include<algorithm>
#include<queue>
#include<deque>
#include<map>
#include<unordered_map>
#include<set>
#include<stack>
//#define LOCAL //提交时一定注释
#include<bits/stdc++.h>
#define VI vector<int>
#define io ios::sync_with_stdio(false); cin.tie(0); cout.tie(0)
using namespace std;
typedef long long LL;
typedef double db;
const int inf = 0x3f3f3f3f;
const LL INF = 1e18;
const int N = 2e5 + 10;
const LL mod = 998244353;
const int p = 26;
inline int readint() {int x; scanf("%d", &x); return x;}
int w[N], fa[N];
int head[N], Next[N << 1], edge[N << 1];
int tot;
int n, m, s, e;
bool vis[N];
bool flag;
void init() {
for(int i = 1; i <= n; ++i) {
fa[i] = i;
head[i] = 0;
}
tot = 0;
}
void add(int u, int v) {
edge[++tot] = v;
Next[tot] = head[u];
head[u] = tot;
}
int find(int x) {return x == fa[x] ? x : fa[x] = find(fa[x]);}
bool dfs(int x, int mid) {
vis[x] = true;
for(int i = head[x]; i; i = Next[i]) {
int k = edge[i];
if (w[x] >= mid && w[k] >= mid) {
return true;
}
if (!vis[k] && dfs(k, mid)) return true;
}
return false;
}
bool dfs2(int x, int mid, int dep) {
if (x == e) {
if (((dep & 1) && w[x] >= mid) || !(dep & 1)) //只需要判断终点的情况
return true;
}
vis[x] = true;
for(int i = head[x]; i; i = Next[i]) {
int k = edge[i];
if (vis[k]) continue;
if (w[x] < mid && w[k] < mid) return false; //因为第一个dfs已经判过两个都大于等于mid的情况,这里只要判断都小于mid的情况。即不满足相邻的两个不同
if (dfs2(k, mid, dep + 1)) return true;
}
return false;
}
bool check(int mid) {
memset(vis, 0, sizeof(vis));
if (dfs(s, mid)) return true;
memset(vis, 0, sizeof(vis));
return dfs2(s, mid, 1);
}
int main() {
#ifdef LOCAL
freopen("input.txt", "r", stdin);
// freopen("output.txt", "w", stdout);
#endif
int t; t = readint();
while (t--) {
init();
scanf("%d%d%d%d", &n, &m, &s, &e);
VI g[n + 1];
int minn = inf, maxx = -1;
for(int i = 1; i <= n; i++) {
w[i] = readint();
maxx = max(maxx, w[i]);
minn = min(minn, w[i]);
}
for(int i = 1; i <= m; i++) {
int x = readint(), y = readint();
add(x, y); add(y, x);
fa[find(x)] = find(y);
}
if (find(s) != find(e)) {
printf("NO\n");
continue;
}
int l = minn, r = maxx;
int ans = 0;
while (l <= r) {
int mid = l + r >> 1;
if (check(mid)) {
ans = mid;
l = mid + 1;
}
else r = mid - 1;
}
if (ans == 0) printf("NO\n"); //答案没有更新
else {
printf("YES\n%d\n", ans);
}
}
return 0;
}
标签:include,return,vis,int,mid,牛客,2021,集训营,true 来源: https://blog.csdn.net/Encore47/article/details/114016647
本站声明: 1. iCode9 技术分享网(下文简称本站)提供的所有内容,仅供技术学习、探讨和分享; 2. 关于本站的所有留言、评论、转载及引用,纯属内容发起人的个人观点,与本站观点和立场无关; 3. 关于本站的所有言论和文字,纯属内容发起人的个人观点,与本站观点和立场无关; 4. 本站文章均是网友提供,不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属;如您发现该文章侵犯了您的权益,可联系我们第一时间进行删除; 5. 本站为非盈利性的个人网站,所有内容不会用来进行牟利,也不会利用任何形式的广告来间接获益,纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。