程序设计思维与实践 Week6 作业 (3/4/数据班)

A - 氪金带东

题意

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

思路

这道题用到了树的直径，将每个电脑看成一个点，由一个点开始bfs，其最长路的另一个端点一定树直径的一个端点，由这个点再进行一次bfs并且记录下距离，其最长路的另一个端点是树直径的另一个端点，所以再从新得到的这个点进行一次bfs，并且取新的距离与上一次bfs的距离的最大值，这就是到这个点最大网线的长度。

总结

主要是树直径的概念，由一个点到达的最长路的端点一定是树直径的一个端点。
代码中主函数的调用三次bfs的过程有一点点冗余。。。

代码

#include<iostream>
#include<string.h>
#include<queue>
#include<vector>
using namespace std;

int dis1[10005];
int dis2[10005];

struct edge
{
	int point;
	int len;
	edge(int a,int b)
	{
		point=a;len=b;
	}
};
vector<edge>* v;
void bfs(int s,int n,int dis[],vector<edge>* v)
{
	queue<int> q;
	q.push(s);
	bool vis[10005];
	memset(vis,0,sizeof(vis));
	vis[s]=1;
	
	while(!q.empty())
	{	
		int cur=q.front();
		q.pop();
		for(int i=0;i<v[cur].size();i++)
		{
			if(vis[v[cur][i].point]==0)
			{
			
				dis[v[cur][i].point]=dis[cur] + v[cur][i].len;
				vis[v[cur][i].point]=1;
				q.push(v[cur][i].point);
			}
		}
	}
}

int main()
{
	int n;
	while(scanf("%d",&n)!=EOF)
	{
		memset(dis1,0,sizeof(dis1));
		vector<edge>* v=new vector<edge>[n+1];
		for(int i=2;i<=n;i++)
		{
			int j,len;
			scanf("%d%d",&j,&len);
		
			v[i].push_back(edge(j,len));
			v[j].push_back(edge(i,len));
		}
		
		bfs(1,n,dis1,v);
		int d1=0,d2=0,maxd=0;
		
		for(int i=1;i<=n;i++)
		{
			if(dis1[i]>maxd)
			{
				maxd=dis1[i];
				d1=i;
			}
		}
		memset(dis1,0,sizeof(dis1));
		bfs(d1,n,dis1,v);
		maxd=0;
		for(int i=1;i<=n;i++)
		{
			if(dis1[i]>maxd)
			{
				maxd=dis1[i];
				d2=i;
			}
		}
		
		memset(dis2,0,sizeof(dis2));
		bfs(d2,n,dis2,v);
		for(int i=1;i<=n;i++)
			cout<<max(dis1[i],dis2[i])<<endl;
		
	}
} 

B - 戴好口罩！

题意

需要尽快找到所有和小A同学直接或者间接接触过的同学，将他们隔离，防止更大范围的扩散。
众所周知，学生的交际可能是分小团体的，一位学生可能同时参与多个小团体内。
请你编写程序解决！
Input
多组数据，对于每组测试数据：
第一行为两个整数n和m（n = m = 0表示输入结束，不需要处理），n是学生的数量，m是学生群体的数量。0 < n <= 3e4 ， 0 <= m <= 5e2
学生编号为0~n-1
小A编号为0
随后，m行，每行有一个整数num即小团体人员数量。随后有num个整数代表这个小团体的学生。
Output
输出要隔离的人数，每组数据的答案输出占一行

思路

这道题可以用并查集的方法来解决，每个人都初始化为一个集合，社团中每次输入的人都与上一个人进行一次合并，并且更新这个集合的人数，最后输出0号所在的那个集合的人数即可，不过要在查找的过程中进行路径紧缩操作，不然会tle。

总结

并查集的应用，对并查集进行了路径紧缩操作

代码

#include<iostream>
using namespace std;

int par[30005],rank[30005];
int find(int n)
{
	if(par[n]==n)
		return n;
	else
		return par[n]=find(par[n]);
}

void unite(int x,int y)
{

	x=find(x);
	y=find(y);
	
	if(x==y)
		return;
	if(rank[x]<rank[y])
		swap(x,y);
	rank[x]+=rank[y];
	par[y]=x;
	
}

int main()
{
	int n,m;
	scanf("%d%d",&n,&m);
	
	while(!(m==0&&n==0))
	{
		for(int i=0;i<n;i++)
		{
			par[i]=i;
			rank[i]=1;
		}
		
		for(int i=0;i<m;i++)
		{
			int cnt;
			int last=-1;
			scanf("%d",&cnt);
			while(cnt--)
			{
				int tmp;
				scanf("%d",&tmp);
				if(last!=-1)
					unite(last,tmp);
		
				last=tmp;
			}
			 
		}	
		cout<<rank[find(0)]<<endl;
		
		scanf("%d%d",&n,&m);
	}
}

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值

思路

这个题是最小生成树算法的应用，我们可以将天也看作是一个点，到其他点的权重是Wi，之后由于边比较多，我们可以采用Prim算法来求最小生成树，将所有的距离dis数组更新为无穷，dis[0] = 0，然后用0号松弛与0号相邻的点，下一步取出dis数组中最小值且不为0的下标的那个点，设置dis=0，并再对相邻点进行松弛操作，重复n-1次，输出中途记录的每次取最小值时的消耗。

总结

这道题是最小生成树算法的应用，将天也看成一个点后直接调用算法即可。

代码

#include<iostream>

using namespace std;

const int MAX=3e2+10;

int g[MAX][MAX];
int cnt=0;
void Prim(int n)
{
	int *dis=new int[n];
	for(int i=0;i<n;i++)
		dis[n]=1e5+5;
	for(int i=0;i<n;i++)
		dis[i]=g[0][i];
	
	for(int i=1;i<n;i++)
	{
		int Min=1e5+5,index=-1;
		for(int j=0;j<n;j++)
		{
			if(dis[j]<Min&&dis[j]!=0)
			{
				index=j;Min=dis[j];
			}
		}
		
		dis[index]=0;
		cnt+=Min;
		for(int j=0;j<n;j++)
		{
			if(dis[j]>(dis[index]+g[index][j]))
				dis[j]=dis[index]+g[index][j];
		}
	}
	
}

int main()
{
	int n;
	scanf("%d",&n);
	g[n][n]=0;
	for(int i=0;i<n;i++)
	{
		int temp;
		scanf("%d",&temp);
		g[n][i]=temp;
		g[i][n]=temp;
	}
	
	for(int i=0;i<n;i++)
		for(int j=0;j<n;j++)
		{
			int temp;
			scanf("%d",&temp);
			g[i][j]=temp;
		}
	
	
	Prim(n+1);
	cout<<cnt<<endl;
}

D - 数据中心

题意

思路

题目比较长，意思就是求边权最大值的最小值，只要求一颗最小生成树然后输出最大边权就可以，这里用的是Kruskal算法来写的，将每个点都初始化为一个集合，并且将边初始化为小根堆，每次取出一条，如果两个顶点不在一个集合中，则合并并且边数加一，直到变数为n-1结束算法。

总结

这道题也可以用二分的方法做，但是感觉求一个最小生成树就可以

代码

#include<iostream>
#include<queue> 
using namespace std;

int cnt=0;
const int Max=5e4+5;
struct edge
{
	int a,b;
	int w;
	edge(int aa,int bb,int cc)
	{
		a=aa;b=bb;w=cc;
	}
	bool operator <(const edge& t) const
	{
		return w>t.w;
	}
};

int par[Max], rnk[Max];

int find(int n)
{
	if(par[n]==n)
		return n;
	else
		return par[n]=find(par[n]);
}

void unit(int x,int y)
{
	x=find(x);
	y=find(y);
	if(x==y)
		return;
	
	if(rnk[x]<rnk[y])
		swap(x,y);
		
	par[y]=x;
	rnk[x]+=rnk[y];
}

void ini(int n)
{
	for(int i=0;i<n;i++)
	{
		par[i]=i;rnk[i]=1;
	}
}


void kur(int m,int n)
{	
	priority_queue<edge> q;
	for(int i=0;i<m;i++)
	{
		int a,b,c;
		scanf("%d%d%d",&a,&b,&c);
		
		q.push(edge(a,b,c));
	}
	
	int e=0;
	while(e<n-1)
	{
		edge tmp=q.top();
		q.pop();
		
		int a=find(tmp.a);
		int b=find(tmp.b);
		if(a!=b)
		{
			unit(a,b);
			e++;
			cnt=max(cnt,tmp.w);
		}
	}
	
}

int main()
{
	int n,m,root;
	scanf("%d%d%d",&n,&m,&root);

	ini(n);
	kur(m,n);
	cout<<cnt<<endl;
}