【洛谷】P1525 关押罪犯（二分+二分图判断）

题目描述

SS城现有两座监狱，一共关押着NN名罪犯，编号分别为1-N1−N。他们之间的关系自然也极不和谐。很多罪犯之间甚至积怨已久，如果客观条件具备则随时可能爆发冲突。我们用“怨气值”（一个正整数值）来表示某两名罪犯之间的仇恨程度，怨气值越大，则这两名罪犯之间的积怨越多。如果两名怨气值为cc 的罪犯被关押在同一监狱，他们俩之间会发生摩擦，并造成影响力为cc的冲突事件。

每年年末，警察局会将本年内监狱中的所有冲突事件按影响力从大到小排成一个列表，然后上报到S 城Z 市长那里。公务繁忙的Z 市长只会去看列表中的第一个事件的影响力，如果影响很坏，他就会考虑撤换警察局长。

在详细考察了NN 名罪犯间的矛盾关系后，警察局长觉得压力巨大。他准备将罪犯们在两座监狱内重新分配，以求产生的冲突事件影响力都较小，从而保住自己的乌纱帽。假设只要处于同一监狱内的某两个罪犯间有仇恨，那么他们一定会在每年的某个时候发生摩擦。

那么，应如何分配罪犯，才能使Z 市长看到的那个冲突事件的影响力最小？这个最小值是多少？

输入格式

每行中两个数之间用一个空格隔开。第一行为两个正整数N,MN,M，分别表示罪犯的数目以及存在仇恨的罪犯对数。接下来的MM行每行为三个正整数a_j,b_j,c_jaj​,bj​,cj​，表示a_jaj​ 号和b_jbj​号罪犯之间存在仇恨，其怨气值为c_jcj​。数据保证1<aj≤bj≤N ,0 < cj≤ 1,000,000,0001<aj≤bj≤N,0<cj≤1,000,000,000，且每对罪犯组合只出现一次。

输出格式

共11 行，为ZZ 市长看到的那个冲突事件的影响力。如果本年内监狱中未发生任何冲突事件，请输出00。

思路：这个题目看网上也有并差集做这个题目的，这里我是用的二分图的判定加上二分枚举答案做的，中间MLE了几发，还是对空间复杂度不是很敏感吧，一开始就是用的邻接数组，但是是MLE了，后来改了结构体就好了

代码：

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<vector>
#define ll long long
using namespace std;
const int maxn=20000+10;
int vis[maxn];
int n,m;
int flag;
struct Node
{
    int to;
    int dist;
    Node(){}
    Node(int a,int b):to(a),dist(b){}
};
vector<Node >G[maxn];
void DFS(int u,int color,int mid)
{
    vis[u]=color;
    for(int i=0;i<G[u].size();i++)
    {
        Node node=G[u][i];
        int v=node.to,dist=node.dist;
        if(dist<=mid) continue;
        if(vis[v]==0)
            DFS(v,3-color,mid);
        else if(vis[v]==color)
        {
            flag=1;
            return ;
        }
    }
}
bool judge(int mid)
{
    memset(vis,0,sizeof(vis));
    flag=0;
    for(int i=1;i<=n;i++)
    {
        if(vis[i]==0)
            DFS(i,1,mid);
    }
    if(flag==0)
        return true;
    else return false;
}
int main()
{
    scanf("%d%d",&n,&m);
    int maxx=0;
    for(int i=1;i<=m;i++)
    {
        int x,y,dist;
        scanf("%d%d%d",&x,&y,&dist);
        G[x].push_back(Node(y,dist));
        G[y].push_back(Node(x,dist));
        maxx=max(maxx,dist);
    }
    int l=0,r=maxx+1;
    int mid;
    int ans=0;
    while(l<=r)
    {
        mid=(l+r)/2;
        if(judge(mid))
        {
            ans=mid;
            r=mid-1;
        }
        else
            l=mid+1;
    }
    printf("%d\n",ans);
}