题目链接
在看了算法进阶指南的二分图后,就对着方面的题跃跃欲试。
这道题目,大佬已经把思路讲的很清楚;
有题目中明显的提示———把罪犯分成两个监狱,在同一个监狱的罪犯之间的怨气值要尽
可能小。那么二分答案判定的东西就出来了。
即在同一个监狱里的罪犯之间最大怨气值
但是这个东西有什么用处呢?
1降低时间复杂度,(好像说了跟没说一样)
2修改原图,删去权值小于等于判定值的边,在进行二分图的判定。
那么另外一个问题又出现了———check函数怎么写。
二分答案无非就两点,二分的东西是什么,现在二分的东西已经解决了,就是同一个监狱里的罪犯之间最大怨气值。
是不是只要修改过后的图是一张二分图就可以了?
答案是yes;
二分图的判定方法常见的有染色法。
以下是李煜东大佬的模板
void dfs(int x,int color){
book[x]=color;
for(int i=head[x]; i ;i=next[i]){
int v=ver[i];
if(!book[v])
dfs(v,3-color);
else if(book[v]==color){
flag=1;
return ;
}
}
}
int flag=0;
for(int i=1;i<=n;i++)
if(!book[i])
dfs(i,1);
现在还需要多说一句。因为二分经常性的就是容易写成死循环。
这里是整数范围内的二分可以套用以下模板;
while(l<r){
int mid=l+r>>1;
if(check(mid))
r=mid;
else l=mid+1;
}
若还不太理解可以参见《算法进阶指南》。
以下是本人的全部代码
#include<bits/stdc++.h>
using namespace std;
const int ll=1e6;
int n,m,tot,flag;
int head[ll],next[ll],ver[ll],wer[ll],d[ll],book[ll];
void add(int u,int v,int w){
tot++;
ver[tot]=v; wer[tot]=w;
next[tot]=head[u];
head[u]=tot;
}
void dfs(int x,int c,int z){
book[x]=c;
for(int i=head[x]; i ;i=next[i]){
int v=ver[i],w=wer[i];
if(w<=z) continue;
if(!book[v])
dfs(v,3-c,z);
else if(book[v]==c){
flag=1;
return ;
}
}
}
bool check(int w){
memset(book,0,sizeof(book));
flag=0;
for(int i=1;i<=n;i++)
if(!book[i]){
dfs(i,1,w);
if(flag) return 0;
}
return 1;
}
int main(){
scanf("%d%d",&n,&m);
for(int i=1;i<=m;i++){
int u,v,w;
scanf("%d%d%d",&u,&v,&w);
add(u,v,w); add(v,u,w);
d[i]=w;
}
sort(d+1,d+m+1);
int l=0,r=d[m];
while(l<r){
int mid=(l+r)/2;
if(check(mid)){
r=mid;
}
else l=mid+1;
}
cout<<r<<endl;
return 0;
}