作业训练2 21. 新型冠状病毒（COVID19）传播

一位学习路上的程序员

已于 2023-08-28 11:23:30 修改

阅读量340

点赞数 2

文章标签：算法

于 2023-08-28 11:16:58 首次发布

本文链接：https://blog.csdn.net/V13594896015/article/details/132533108

版权

【问题描述】

在一个风景秀丽的小镇，一天早上，有 N 名晨跑爱好者（编号 1 ～ N ）沿着优雅的江边景观道朝同一方向进行晨跑，第 i 名跑者从位置 Si 处起跑，且其速度为 Vi。换句话说，对所有的实数 t ≥ 0，在时刻 t 时第 i 名跑者的位置为 Si + Vi ·t。

很不幸的是，其中一名跑者在 t = 0 的时刻感染了病毒，且是无症状感染者，这种病毒只会在同一时刻处在同一位置的跑者之间传播，新感染了病毒的跑者也会感染其他人，很显然，等待足够长的时间，那么病毒会感染一些特定的跑者。

事后发现其中有一名跑者感染了新冠病毒，如果此人就是在 t = 0 时刻的那名感染者，那么，在 N 名晨跑爱好者中会有多少人感染新冠病毒？

【输入形式】

输入包含三行：

- 第一行包含为两个整数 N 和 K，分别表示运动员的人数以及开始时感染了病毒的跑者编号。
- 第二行包含 N 个正整数 S1、S2、...、SN，用空格隔开，分别表示跑者的起始位置。
- 第三行包含 N 个正整数 V1、V2、...、VN，用空格隔开，分别表示跑者的速度。

【输出形式】

输出为一个整数，表示最终被感染人数。

【样例输入】

6 3
3 9 8 5 7 5
6 6 5 4 6 3

【样例输出】

【样例说明】
【评分标准】

对于50%的评测用例，0< K ≤ N ≤10^2

对于70%的评测用例，0< K ≤ N ≤10^4

对于90%的评测用例，0< K ≤ N ≤10^6

对于100%的评测用例，0< K ≤ N ≤10^7

解题思路：

不知道大家的想法，这道题我一开始自己做的时候是从正面开始想的，大家可以思考一下，从感染源出发，最多只能二次感染绝对不会发生第三次感染！（这很关键！），图示如下：

代码如下：

#include<iostream>
#include<vector>
using namespace std;
int main()
{
	//1.输入
	int n, k;
	cin >> n >> k;
	int x0, v0;
	vector<int> addx;//储存初始位置
	vector<int> addv;//储存初始速度
	vector<int> add1;//记录是否感染
	for (int i = 0; i < n; i++)
	{
		int x;
		cin >> x;
		addx.push_back(x);
		add1.push_back(0);
		if (i == k - 1)//源头
		{
			x0 = x;
		}
	}
	for (int i = 0; i < n; i++)
	{
		int v;
		cin >> v;
		addv.push_back(v);
		if (i == k - 1)//源头
		{
			v0 = v;
		}
	}
	//按位置排序
	for (int i = 0; i < n; i++)
	{
		for (int j = n - 1; j > 0; j--)
		{
			if (addx[j] < addx[j - 1])
			{
				swap(addx[j], addx[j - 1]);
				swap(addv[j], addv[j - 1]);
			}
		}
	}
	//记录交换后k的位置
	for (int i = 0; i < n; i++)
	{
		if (addx[i] == x0 && addv[i] == v0)
		{
			k = i;
			break;
		}
	}

	//2.第一次感染

	vector<int> add;//记录感染者的编号
	long num=0;//记录感染者人数,初始为0!!!
	for (int i = 0; i < n; i++)
	{
		if (addx[i]==addx[k])
		{
			add.push_back(i);
			add1[i] = 1;
			num++;
		}
		else if (i < k&&addv[i]>addv[k])
		{
			add.push_back(i);
			add1[i] = 1;
			num++;
		}
		else if (i ==k)
		{
			add.push_back(k);
			add1[i] = 1;
			num++;//感染自己
		}
		else if (i > k && addv[i] < addv[k])
		{
			add.push_back(i);
			add1[i] = 1;
			num++;
		}
	}

	//3.第二次感染
	int w = num;
	for (int i = 0; i < w; i++)
	{
		if (num == n)
		{
			break;
		}
		if (add[i] == k)
		{
			continue;
		}
		else if (add[i] < k)
		{
			for (int j = k + 1; j < n; j++)
			{
				if (addv[add[i]] > addv[j]&&add1[j]==0)
				{
					add1[j] = 1;
					num++;
				}
			}
		}
		else if (add[i] > k)
		{
			for (int j = k - 1; j >= 0; j--)
			{
				if (addv[add[i]] < addv[j]&&add1[j]==0)
				{
					add1[j] = 1;
					num++;
				}
			}
		}
	}

	//4.输出
	cout << num;
	system("pause");
	return 0;
}

但是这种做法最后三个案列运行时间过长(因为这种方法有n^2---冒泡排序和找到二次感染者），我又没想到好的方法提速，因此参考了一下网上的做法，发现可以从反面来想，即找到安全的人而不是找到被感染的人！！！这种算法只需一次循环!!!非常高效，这里引用了一下一位博客的思路：

原文链接：https://blog.csdn.net/Karltan/article/details/118672419

解题思路
不得不说，这是训练二中最难的一道题，也是一道追及问题，而作为理科生的我们，肯定都知道物理老师曾经说过的 x-t 图法，但是这道题并不是用 x-t 图法做，而是用于理解题意。

话不多说，上图！

显然，根据高中物理，①-⑥中，有③和④最终会被感染。但是因为找出所有感染者的思路较难，所以我们反向寻找，找到安全的人的数目。

而安全的人有什么特点呢？

我们将安全的人分为两个部分：一部分人初始位置小于零号感染者；另一部分人初始位置大于零号感染者。

初始位置小于零号感染者的人要想不被感染，其速度只能小于等于右边所有人的最小速度。因为我们将零号感染者看做在右边。右边速度最小的那个人一定会被零号感染者感染。而左边人的速度要是大于这个最小速度，那么也会被感染。

所以初始位置大于零号感染者的人要想不被感染，其速度只能大于等于左边所有人的最大速度。

所以我们以零号感染者作为分界后，找出左边的最大速度，右边的最小速度。

遍历右边时，如果速度大于该速度且位置大于零号感染者的位置，我们将计数器+1。

再遍历左边，遍历时如果速度小于右边的最小速度且位置小于零号感染者的位置，将计数器+1。

自己写了一下代码：

#include<iostream>
#include<vector>
using namespace std;
int main()
{
	//1.输入
	int n, k; cin >> n >> k;
	vector<int > adds;//记录初始位置
	vector<int >addv;//记录初始速度
	int s0, v0;//记录源头的位置和速度
	for (int i = 0; i < n; i++)
	{
		int temp; cin >> temp;
		adds.push_back(temp);
		if (i == k - 1)
		{
			s0 = temp;
		}
	}
	for (int i = 0; i < n; i++)
	{
		int temp; cin >> temp;
		addv.push_back(temp);
		if (i == k - 1)
		{
			v0 = temp;
		}
	}

	//2.不能按位置给数据排序,n*logn也很大
	//3.记录此时感染者的位置为x
	int x=k-1;
	//4.找到左边的最大速度和右边的最小速度，包括源头
	int vmax=addv[x], vmin=addv[x];
	for (int i = 0; i <n; i++)
	{
		if (adds[i] <= s0)
		{
			if (vmax < addv[i]) vmax = addv[i];
		}
		if (adds[i] >= s0)
		{
			if (vmin > addv[i]) vmin = addv[i];
		}
	}

	//5.找到安全的人数,注意取等时的特殊情况
	int num = 0;
	for (int i = 0; i < n; i++)
	{
		//左边的
		if (adds[i] < s0)
		{
			if (addv[i] < vmin) num++;
			else if (addv[i] == vmin && adds[i] < s0)
				num++;
		}
		//右边的
		if (adds[i] > s0)
		{
			if (addv[i] > vmax) num++;
			else if (addv[i] == vmax && adds[i] > s0)
				num++;
		}
	}
	
	//6.输出感染人数
	cout << (n - num) << endl;
	system("pause");
	return 0;
}

------一位学习路上的程序员