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先从任意一个端点开始,找到距离它最远的端点,就是两个端点中的一个。
在用dfs从得到的这个端点开始,得到它距其他所有点的距离。同时离它最远的点就是另外一个端点。
再从另外一个端点dfs一遍,输出这两组距离中最大的即可。
反思
最开始受题中图片的限制,认为是一条链的,因此写的就是在一条链里面寻找最长网线,在经过WA的警告后,再次审题明白这是一棵普通的树。
之后又得到了ML的警告,是因为每组数组新开始的时候需要清空vector,否则,vector一直增加会超内存。
借鉴
利用vector的想法,借鉴于:
代码
#include<iostream>
#include<vector>
#include<stdio.h>
using namespace std;
/*A-氪金带东 */
vector<pair<int,int>> edge[100010];//边集合
int len[100010] = { 0 };//记录首尾端点到其他各个端点的最大距离
int maxlen = 0;//记录最长的距离,根据此距离得到首尾端点
int first;//寻找首尾端点
void pathLength(int u,int v,int length)
{
if (maxlen < length)
{
maxlen = length;
first = u;
}
int size = (int)edge[u].size();
for (int i = 0; i < size; i++)
{
int tempv = edge[u][i].first;
int tempw = edge[u][i].second;
if (v == tempv)
{
continue;
}
pathLength(tempv, u, length + tempw);
if (tempw+length >=len[tempv])//更新长度
{
len[tempv] = length + tempw;
}
}
}
int main()
{
int n;
while (scanf("%d",&n) != EOF)
{
maxlen = 0;
for (int i = 0; i <= n; i++)
{
len[i] = 0;
}
memset(edge, 0, sizeof(edge));
for (int i = 2; i <= n; i++)
{
int v, w;
//cin >> v >> w;
scanf("%d %d", &v, &w);
pair<int,int> p(v, w);
edge[i].push_back(p);
p.first = i;
edge[v].push_back(p);
}
pathLength(2, -1, 0);//寻找一个直径最长的端点
pathLength(first, -1, 0);//得到该端点到其他点的距离,并得到另外一个端点
pathLength(first, -1, 0);//得到两个端点到其他点的最大距离
for (int i = 1; i <= n; i++)
{
printf("%d\n", len[i]);
//cout << len[i] << endl;
}
}
}
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
思路
这道题采用并查集的思路,初始每个个体是一个集合,之后每次有一个小团体出现,那么该小团体的人是一个集合,并且如果该集合里面有人属于另外的集合,那么这两个集合合并。
同时记录每一个集合里面的人数。
注意:并查集find查找时,要路径压缩!!!呜呜呜这个真的惨!!尤其是D题!!血与泪的教训!! 听我的好吗?一定要路径压缩!!好吗?压缩它! |
---|
代码
#include<iostream>
#include<vector>
#include<stdio.h>
using namespace std;
/*B-戴好口罩*/
int n, m;//学生数量、学生群体数量
int from[30010] = { -1 };
int fsize[30010] = { 1 };
void init()
{
//初始化
for (int i = 0; i < n; i++)
{
from[i] = i;
fsize[i] = 1;
}
}
int find(int x)
{//寻找x属于哪个集合
if (from[x] == x)
{
return x;
}
return from[x]=find(from[x]);//就是这里!压缩它好吗?听我的!
}
void merge(int x, int y)//合并两个元素
{
x = find(x);
y = find(y);
if (x != y)//不在同一个集合里面合并
{
from[y] = x;
fsize[y] = fsize[y] + fsize[x];
fsize[x] = fsize[y];
}
}
int main()
{
while (1)
{
cin >> n >> m;
if (n == 0 && m == 0)//结束标志
{
break;
}
init();//初始化
for (int i = 0; i < m; i++)
{
int num;//小团体人数
cin >> num;
int x, y;
cin >> x;
for (int j = 1; j < num; j++)
{
cin >> y;
merge(x, y);
x = y;
}
}
cout << fsize[find(0)] << endl;
}
}
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值
Sample Input
4
5
4
4
3
0 2 2 2
2 0 3 3
2 3 0 4
2 3 4 0
Sample Output
9
思路
这道题可以看作连通图中求最小生成树的问题。
把农田看作点,两块农田连接时需要消耗的MP看作边的权重,黄河之水天上来的点可以看作是超级源点。
据此,存储图的各个边,利用Krusal算法来生成最小生成树,使用并查集来判断是否会成环(即边的两个顶点是否在一个集合里。)
将边按照权重从小到大排序,依次选出最小的边来进行计算。
每次加上选取的边的权重即可。
ps:还记得我上一题说了什么嘛?对的就是压缩它!压缩路径,呜呜呜,一定要记好哦! |
---|
代码
#include<iostream>
#include<algorithm>
#include<vector>
#include<stdio.h>
using namespace std;
/*C-掌握魔法の东东 I */
//连通图中求最小生成树问题
struct weightedEdge//边结构体
{
int v1;
int v2;
int w;
};
bool cmp(weightedEdge& a, weightedEdge& b)//排序方法
{
return a.w < b.w;
}
weightedEdge edge[100010];
int edgesize = 0;//边的个数
int from[100010];//用作并查集
int n;//农田数量
int pointnumber = 0;//最小生成树里面农田的数量
int minMP;//消耗最小的minMP
int find(int x)
{//寻找x属于哪个集合
if (from[x] == x)
{
return x;
}
return from[x]=find(from[x]);
}
void merge(int x, int y,int i)//合并两个元素
{
x = find(x);
y = find(y);
if (x != y)//不在同一个集合里面合并
{
from[y] = x;
pointnumber++;//选择的点的农田个数增加
minMP += edge[i].w;//选择这条边
}
}
void pushbackEdge(int v1, int v2, int w)//添加边
{
edge[edgesize].v1 = v1;
edge[edgesize].v2 = v2;
edge[edgesize].w = w;
edgesize++;
}
int main()
{
cin >> n;
int wi;
minMP = 0;
for (int i = 1; i <= n; i++)
{
cin >> wi;
pushbackEdge(0, i, wi);
from[i] = i;
}
int pij;
for (int i = 1; i <= n; i++)
{
for (int j = 1; j <= n; j++)
{
cin >> pij;
if (i == j)
{
continue;
}
pushbackEdge(i, j, pij);
}
}
//按边的权重升序排序
sort(edge, edge + edgesize, cmp);
//求最小生成树
for (int i = 0; i < edgesize; i++)
{
int v1 = edge[i].v1;
int v2 = edge[i].v2;
merge(v1, v2, i);
if (pointnumber == n)
{
break;
}
}
cout << minMP << endl;
}
D - 数据中心
Sample Input
4
5
1
1 2 3
1 3 4
1 4 5
2 3 8
3 4 2
Sample Output
4
Note
思路
其实这道题与上一题类似,不过这道题是在求最小生成树中最大的边值。
因此只需要修改部分就可以。
知道我要问什么了么?今天你路径压缩了么? 没有?赶快压缩!!嘤!! |
---|
代码
#include<iostream>
#include<algorithm>
#include<stdio.h>
using namespace std;
/*D-数据中心*/
struct weightedEdge//边结构体
{
int v1;
int v2;
int w;
bool operator<(const weightedEdge& rv)const
{
return w < rv.w;
}
};
//bool cmp(weightedEdge& a, weightedEdge& b)//排序方法
//{
// return a.w < b.w;
//}
weightedEdge edge[100010];
int edgesize = 0;//边的个数
int from[100010];//用作并查集
int n, m, root;//节点数、边数、根节点
int pointnumber = 0;//最小生成树里面节点的数量
int Tmax = 0;//树结构传输时间
int find(int x)
{//寻找x属于哪个集合
if (from[x] == x)
{
return x;
}
return from[x] = find(from[x]);
}
void pushbackEdge(int v1, int v2, int w)//添加边
{
edge[edgesize].v1 = v1;
edge[edgesize].v2 = v2;
edge[edgesize].w = w;
edgesize++;
}
int main()
{
scanf("%d", &n);
scanf("%d", &m);
scanf("%d", &root);
for (int i = 1; i <= n; i++)
{//初始化
from[i] = i;
}
int v1, v2, w;
for (int i = 0; i < m; i++)
{
scanf("%d %d %d", &v1, &v2, &w);
pushbackEdge(v1, v2, w);
}
//按边的权重升序排序
sort(edge, edge + edgesize);
//求最小生成树并得到Tmax
for (int i = 0; i < edgesize; i++)
{
v1 = find(edge[i].v1);
v2 = find(edge[i].v2);
if (v1 != v2)//不在同一个集合里面合并
{
from[v1] = v2;
pointnumber++;//选择的点的农田个数增加
/*if (edge[i].w > Tmax)
{
Tmax = edge[i].w;
}*/
Tmax = max(Tmax, edge[i].w);
}
if (pointnumber == n-1)
{
break;
}
}
printf("%d\n", Tmax);
}