Program week 6 work

A - 氪金带东

实验室里原先有一台电脑(编号为1),最近氪金带师咕咕东又为实验室购置了N-1台电脑,编号为2到N。每台电脑都用网线连接到一台先前安装的电脑上。但是咕咕东担心网速太慢,他希望知道第i台电脑到其他电脑的最大网线长度,但是可怜的咕咕东在不久前刚刚遭受了宇宙射线的降智打击,请你帮帮他。

在这里插入图片描述

提示: 样例输入对应这个图,从这个图中你可以看出,距离1号电脑最远的电脑是4号电脑,他们之间的距离是3。 4号电脑与5号电脑都是距离2号电脑最远的点,故其答案是2。5号电脑距离3号电脑最远,故对于3号电脑来说它的答案是3。同样的我们可以计算出4号电脑和5号电脑的答案是4.
Input
输入文件包含多组测试数据。对于每组测试数据,第一行一个整数N (N<=10000),接下来有N-1行,每一行两个数,对于第i行的两个数,它们表示与i号电脑连接的电脑编号以及它们之间网线的长度。网线的总长度不会超过10^9,每个数之间用一个空格隔开。
Output
对于每组测试数据输出N行,第i行表示i号电脑的答案 (1<=i<=N).
Sample Input
5
1 1
2 1
3 1
1 1
Sample Output
3
2
3
4
4

思路

本题若暴力求解,即直接对每个点dfs,但是复杂度为O(n*(n+m)),n为点数,m为边数,复杂度太高。所以需要转换思路:
该题需要用到树的直径(任意两个节点之间的最大距离)
求直径的方法:由任意一个顶点开始搜索,找到距离最远的节点v1。再从v1开始搜索,找到距离v1最远的节点v2,v1,v2就是直径的两个端点。搜索可用dfs也可以用bfs。
所以对于本题,每个点到任意一个点的最大距离即为dis(v1,i)或dis(v2,i),同时本题采用链式前向星存储边,先对节点1进行dfs1,求出v1,再对v1进行dfs2,再对v2进行dfs3,然后取每个点dfs2和dfs3的最大值,即为答案。

Answer

具体思路见注释

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<string.h>
using namespace std;
const int maxn=1e5+100;
struct Edge{
	int v,w,next;
}edge[2*maxn];
int head[maxn],tot,ans1,ans2,ans3,v1,v2;
int dis1[maxn],dis2[maxn],dis3[maxn];
bool vis1[maxn],vis2[maxn],vis3[maxn];
void init(int n){
	memset(head,-1,sizeof(head));
	memset(vis1,0,sizeof(vis1));
	memset(vis2,0,sizeof(vis2));
	memset(vis3,0,sizeof(vis3));
	memset(dis1,0,sizeof(dis1));
	memset(dis2,0,sizeof(dis2));
	memset(dis3,0,sizeof(dis3));
	ans1=0,ans2=0,ans3=0,tot=0;
}
void add(int u,int v,int w){
	tot++;
	edge[tot].v=v;
	edge[tot].w=w;
	edge[tot].next=head[u];
	head[u]=tot;
} 
void dfs1(int u){
	vis1[u]=1;
	for(int i=head[u];i!=-1;i=edge[i].next){
		int v=edge[i].v;
		if(!vis1[v]){
			vis1[v]=1;
			dis1[v]=dis1[u]+edge[i].w;
			if(ans1<dis1[v]){
				ans1=dis1[v];
				v1=v;
			}
			dfs1(v);
		}
	}
}

void dfs2(int u){
	vis2[u]=1;
	for(int i=head[u];i!=-1;i=edge[i].next){
		int v=edge[i].v;
		if(!vis2[v]){
			vis2[v]=1;
			dis2[v]=dis2[u]+edge[i].w;
			if(ans2<dis2[v]){
				ans2=dis2[v];
				v2=v;
			}
			dfs2(v);
		}
	}
}
void dfs3(int u){
	vis3[u]=1;
	for(int i=head[u];i!=-1;i=edge[i].next){
		int v=edge[i].v;
		if(!vis3[v]){
			vis3[v]=1;
			dis3[v]=dis3[u]+edge[i].w;
			if(ans3<dis3[v]){
				ans3=dis3[v];
			}
			dfs3(v);
		}
	}
}

int main(){
	int N;
	while(cin>>N){
		init(N);
		for(int i=2;i<=N;i++){
			int v,w;
			cin>>v>>w;
			add(i,v,w);
			add(v,i,w);
		}
	dfs1(1);
	dfs2(v1);
	dfs3(v2);
	for(int i=1;i<=N;i++)cout<<max(dis2[i],dis3[i])<<endl;	
	}
	return 0;
}

B - 戴好口罩!

新型冠状病毒肺炎(Corona Virus Disease 2019,COVID-19),简称“新冠肺炎”,是指2019新型冠状病毒感染导致的肺炎。
如果一个感染者走入一个群体,那么这个群体需要被隔离!
小A同学被确诊为新冠感染,并且没有戴口罩!!!!!!
危!!!
时间紧迫!!!!
需要尽快找到所有和小A同学直接或者间接接触过的同学,将他们隔离,防止更大范围的扩散。
众所周知,学生的交际可能是分小团体的,一位学生可能同时参与多个小团体内。
请你编写程序解决!戴口罩!!

Input
多组数据,对于每组测试数据:
第一行为两个整数n和m(n = m = 0表示输入结束,不需要处理),n是学生的数量,m是学生群体的数量。0 < n <= 3e4 , 0 <= m <= 5e2
学生编号为0~n-1
小A编号为0
随后,m行,每行有一个整数num即小团体人员数量。随后有num个整数代表这个小团体的学生。

Output
输出要隔离的人数,每组数据的答案输出占一行

Sample Input
100 4
2 1 2
5 10 13 11 12 14
2 0 1
2 99 2
200 2
1 5
5 1 2 3 4 5
1 0
0 0

Sample Output
4
1
1

思路

并查集的例题,题目给的是多个小团体,多个团体的人都加入并查集,最后对所有人进行遍历,看是否在一个并查集中(根节点相同)

Answer

具体思路见注释

#include<iostream>
using namespace std;
const int maxn=5e5+100;
int n,m,sum,ans,father[maxn];
int find(int x)
{
	if(father[x]!=x)
	father[x]=find(father[x]);
	return father[x];
}
void Union(int a,int b)
{
	int fa=find(a);
	int fb=find(b);
	father[fa]=fb;
}
int main()
{
	int a,b;
	cin>>n>>m; 
	while(!(n==0&&m==0))
	{
		ans=0;
		for(int i=0;i<n;i++)
		father[i]=i; 
		for(int i=1;i<=m;i++)
		{
			cin>>sum;
			cin>>a;
			a=find(a);
			for(int j=2;j<=sum;j++)
			{
				cin>>b;
				b=find(b);
				if(a!=b)
				Union(a,b);
			}
		}
		int f0=find(0);
		for(int i=0;i<n;i++){
			if(find(i)==f0)
			ans++;
		}
		cout<<ans<<endl;
		cin>>n>>m; 
	}
	return 0;
}

C - 掌握魔法の东东 I

东东在老家农村无聊,想种田。农田有 n 块,编号从 1~n。种田要灌氵
众所周知东东是一个魔法师,他可以消耗一定的 MP 在一块田上施展魔法,使得黄河之水天上来。他也可以消耗一定的 MP 在两块田的渠上建立传送门,使得这块田引用那块有水的田的水。 (1<=n<=3e2)
黄河之水天上来的消耗是 Wi,i 是农田编号 (1<=Wi<=1e5)
建立传送门的消耗是 Pij,i、j 是农田编号 (1<= Pij <=1e5, Pij = Pji, Pii =0)
东东为所有的田灌氵的最小消耗

Input
第1行:一个数n
第2行到第n+1行:数wi
第n+2行到第2n+1行:矩阵即pij矩阵

Output
东东最小消耗的MP值

Example
Input
4
5
4
4
3
0 2 2 2
2 0 3 3
2 3 0 4
2 3 4 0
Output
9

思路

所谓黄河之水天上来,其实就是多一个节点罢了。所以引入0号节点,并在边集中加入0号节点到所有节点的边,之后就是kruskal生成最小生成树

Answer

具体思路见注释

#include<iostream>
#include<algorithm>
using namespace std;
const int maxn=1e5+100;
int n,w,tot=1,sum,ans,father[maxn];
struct node
{
	int u,v,w;
	bool operator <(const node&p)
	{
		return w<p.w;
	}
}edge[maxn];

int find(int x)
{
	if(father[x]!=x)
	father[x]=find(father[x]);
	return father[x];
}

void Union(int a,int b)
{
	int fa=find(a);
	int fb=find(b);
	father[fb]=fa;
}

int main()
{
	cin>>n;
	for(int i=1;i<=n;i++)
	{
		cin>>w;
		edge[tot].u=0;
		edge[tot].v=i;
		edge[tot].w=w;
		tot++; 
	}
	for(int i=1;i<=n;i++){
	    for(int j=1;j<=n;j++){
	  	    cin>>w;
	  	    if(i!=j)
	  	    {
	  		    edge[tot].u=i;
		  	    edge[tot].v=j;
		  	    edge[tot].w=w;
		  	    tot++;
		    }
	    }
	}
	tot--;
	for(int i=1;i<=n+1;i++)
	father[i]=i;
	sort(edge+1,edge+tot+1);
	for(int i=1;i<=tot&&sum<n;i++)
	{
		if(find(edge[i].u)!=find(edge[i].v))
		{
			Union(edge[i].u,edge[i].v);
			ans+=edge[i].w;
			sum++;
		}
	}
	cout<<ans;
	return 0;
}

D - 数据中心

在这里插入图片描述
Input
4
5
1
1 2 3
1 3 4
1 4 5
2 3 8
3 4 2
Output
4
在这里插入图片描述

思路

"root节点只能接收数据,其余任何一个节点可以将数据传输给另外的一个节点,但是不能将数据传输给多个节点"其实是在拐弯抹角告诉你这是一棵树,root节点就是树的根。
这里引入瓶颈生成树的定义:对于一个无向图G,其瓶颈生成树是一棵 “ 树的最大边权值在G的所有生成树中最小 ” 的生成树(这样的生成树可能不止一棵)。
本题即为寻找这样一棵瓶颈生成树,而最小生成树一定是一棵瓶颈生成树。要获得最优的耗时,就是要让Tmax最小,所以目的就是找到一棵最小生成树,该最小生成树的最大边即为答案。

Answer

具体思路见注释

#include<iostream>
#include<cstdio>
#include<algorithm>
using namespace std;
const int maxn=1e5+100;
int n,m,tot=0,root,ans,father[maxn];

struct node
{
	int u,v,w;
	bool operator <(const node&p)
	{
		return w<p.w;
	}
}edge[maxn];

int find(int x)
{
	if(father[x]!=x)
	father[x]=find(father[x]);
	return father[x];
}

void Union(int a,int b)
{
	int fa=find(a);
	int fb=find(b);
	father[fb]=fa;
}

int main()
{
	cin>>n>>m>>root;
	for(int i=1;i<=m;i++)cin>>edge[i].u>>edge[i].v>>edge[i].w;
	for(int i=1;i<=n;i++)father[i]=i;
	sort(edge+1,edge+m+1);
	for(int i=1;i<=m&&tot<n-1;i++){
		if(find(edge[i].u)!=find(edge[i].v)){
			Union(edge[i].u,edge[i].v);
			ans=max(ans,edge[i].w);
			tot++;
		}
	}
	cout<<ans;
	return 0;
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值