萌新求助,40pts
查看原帖
萌新求助,40pts
107236
Zesty_Fox楼主2021/3/31 21:48

虽然在某大佬的帮助下过了此题,但是依然对原来的错解有一些疑问。

错解:(40 pts)提交记录

#include <iostream>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <cmath>
#include <ctime>

#include <algorithm>
#include <vector>
#include <map>
#include <set>
#include <queue>
using namespace std;

typedef long long ll;
typedef vector<int> vi;
typedef pair<int,int> pii;

template<typename T>
inline T read(){
    T x=0,f=1;char ch=getchar();
    while(ch<'0'||ch>'9'){
        if(ch=='-') f=-1;
        ch=getchar();
    }
    while(ch>='0'&&ch<='9'){
        x=(x<<1)+(x<<3)+(ch^48);
        ch=getchar();
    }
    return x*f;
}

#define rdi read<int>
#define rdl read<ll>
#define fi first
#define sec second
#define pb push_back
#define mp make_pair

const int N=1000010;
int n,m,need,tot;
struct Edge{
    int fr,to,w,col;
    bool operator<(const Edge& a)const{
        return w==a.w?col<a.col:w<a.w;
    }
}e[N];
void addedge(int x,int y,int w,int col){
    e[++tot]=(Edge){x,y,w,col};
}

int f[N];
int find(int x){return x==f[x]?f[x]:f[x]=find(f[x]);}
bool merge(int x,int y){
    x=find(x),y=find(y);
    if(x==y) return 0;
    f[y]=x;return 1;
}

bool check(int val){
    for(int i=1;i<=n;i++) f[i]=i;
    for(int i=1;i<=tot;i++){
        if(!e[i].col) e[i].w+=val;
    }
    sort(e+1,e+tot+1);
    int cnt=0,cntw=0;
    for(int i=1;i<=tot;i++){
        int x=e[i].fr,y=e[i].to;
        if(merge(x,y)) cntw+=(!e[i].col),cnt++;
        if(cnt==n-1){
            for(int i=1;i<=tot;i++){
                if(!e[i].col) e[i].w-=val;
            }
            return cntw>=need;
        }   
    }
}

int main(){
    n=rdi(),m=rdi(),need=rdi();
    for(int i=1;i<=m;i++){
        int x=rdi()+1,y=rdi()+1,w=rdi(),col=rdi();
        addedge(x,y,w,col);
    }
    int l=-200,r=200;
    while(l<r){
        int mid=(l+r+1)>>1;
        if(check(mid)) l=mid;
        else r=mid-1;
    }
    for(int i=1;i<=n;i++) f[i]=i;
    for(int i=1;i<=tot;i++){
        if(!e[i].col) e[i].w+=l;
    }
    sort(e+1,e+tot+1);
    int cnt=0,cntw=0,sum=0;
    for(int i=1;i<=tot;i++){
        int x=e[i].fr,y=e[i].to;
        if(!e[i].col&&cntw==need) continue;
        if(merge(x,y)) cntw+=(!e[i].col),cnt++,sum+=e[i].w;
        if(cnt==n-1) break;
    }
    cout<<sum-need*l<<endl;
    return 0;
}

后来把这一行注释掉了,就过了:(倒数第7行)

//if(!e[i].col&&cntw==need) continue;

求教,为什么这样判断刚好选 K 条白边会错?还是数据有问题?

(本来有一个讨论与我这个相关,但那里面基本都是无意义内容。)

2021/3/31 21:48
加载中...