国王饮水记 题解(

题目传送门
step1:暴力
对于每次询问,都进行一次krusal计算。

#include<cstdio>
#include<algorithm>
#define maxn 5039
using namespace std;
inline int read(){
    
    
	char c=getchar();
	int sum=0,flag=0;
	while((c<'0'||c>'9')&&c!='-') c=getchar();
	if(c=='-')
		c=getchar(),flag=1;
	while('0'<=c&&c<='9'){
    
    
		sum=(sum<<1)+(sum<<3)+(c^48);
		c=getchar();
	}
	if(flag) return -sum;
	return sum;
}
struct JTZ{
    
    
	int f,t,w;
	bool operator < (const JTZ x) const {
    
    
		return this->w < x.w;
	}
}e[maxn];
int f[maxn];
int n,m;
int getfa(int x){
    
    
	if(x==f[x]) return x;
	return f[x]=getfa(f[x]);
}
void getans(int x){
    
    
	int sum=0;
	for(int i=1;i<=n;i++) f[i]=i;
	for(int i=1;i<=x;i++){
    
    
		if(getfa(e[i].f)!=getfa(e[i].t)){
    
    
			f[getfa(e[i].f)]=getfa(e[i].t);
			sum+=e[i].w;
		}
	}
	int tmp=getfa(1),ans=0;
	for(int i=1;i<=n;i++)
	    if(getfa(i)==tmp)
	        ans++;
	printf("%d %d\n",sum,ans);
}
int main(){
    
    
	//freopen("1.in","r",stdin);
    n=read(); m=read();
    for(int i=1;i<=n;i++)
        f[i]=i;
    for(int i=1;i<=m;i++){
    
    
    	e[i].f=read();
    	e[i].t=read();
    	e[i].w=read();
        sort(e+1,e+i+1);
    	getans(i);
	}
	return 0;
}

step 2:优化
将排序改为 O ( l o g N + N ) O\left(log_N+N\right) O(logN+N) 的二分查找,但是分数怎么变少了啊

#include<cstdio>
#include<algorithm>
#define maxn 5039
using namespace std;
inline int read(){
    
    
	char c=getchar();
	int sum=0,flag=0;
	while((c<'0'||c>'9')&&c!='-') c=getchar();
	if(c=='-')
		c=getchar(),flag=1;
	while('0'<=c&&c<='9'){
    
    
		sum=(sum<<1)+(sum<<3)+(c^48);
		c=getchar();
	}
	if(flag) return -sum;
	return sum;
}
struct JTZ{
    
    
	int f,t,w;
}e[maxn];
int f[maxn];
int n,m;
int getfa(int x){
    
    
	if(x==f[x]) return x;
	return f[x]=getfa(f[x]);
}
void getans(int x){
    
    
	int sum=0;
	for(int i=1;i<=n;i++) f[i]=i;
	for(int i=1;i<=x;i++){
    
    
		if(getfa(e[i].f)!=getfa(e[i].t)){
    
    
			f[getfa(e[i].f)]=getfa(e[i].t);
			sum+=e[i].w;
		}
	}
	int tmp=getfa(1),ans=0;
	for(int i=1;i<=n;i++)
	    if(getfa(i)==tmp)
	        ans++;
	printf("%d %d\n",sum,ans);
}
int main(){
    
    
	//freopen("1.in","r",stdin);
    n=read(); m=read();
    for(int i=1;i<=n;i++)
        f[i]=i;
    for(int i=1;i<=m;i++){
    
    
    	e[i].f=read();
    	e[i].t=read();
    	e[i].w=read();
    	int l,r,mid;
    	l=1,r=i;
    	while(l+1<r){
    
    
    		mid=l+r>>1;
    		if(e[mid].w<=e[i].w) l=mid;
    		else r=mid;
		}
		JTZ tmp=e[i];
		for(int j=i-1;j>=l+1;j--)
		    e[j+1]=e[j];
		e[l+1]=tmp;
    	getans(i);
	}
	return 0;
}

step3:未完成
将输入的一条边进行判断:
如果两点不连通,那么就连起来。
如果两条边连起来能减小生成树的边权值,那就连起来,否则就不连。
代码略。

猜你喜欢

转载自blog.csdn.net/jiangtaizhe/article/details/108947453