新冠病毒的模拟传播 !

在一个风景秀丽的小镇,一天早上,有 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
【样例输出】

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

 对于50%的评测用例,0< KN102

 对于70%的评测用例,0< KN104

 对于90%的评测用例,0< KN106

 对于100%的评测用例,0< KN107

#include<iostream>
#include<cmath>
#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
ll max_v = LONG_MIN; ll min_v = LONG_MAX;
const int MAXSIZE = 1e7 + 10;

ll s[MAXSIZE];
ll v[MAXSIZE];
int main()
{
    ll n,k;
    cin>>n>>k;
    for(int i = 1 ; i< n + 1 ; i++)
    {
        cin>>s[i];
    }
    for(int j = 1 ; j < n + 1 ; j ++)
    {
        cin>>v[j];
    }
    int ans = 0;
    for(int i = 1 ; i <= n ; i ++)
    {
        //must be infected
        if(s[i] < s[k] && v[i] > v[k] || s[i] > s[k] && v[i] < v[k])
        {
            max_v = max(max_v,v[i]);
            min_v = min(min_v,v[i]);
        }
        if(s[i] == s[k])
        {
            ans++;
            max_v = max(max_v,v[i]);
            min_v = min(min_v,v[i]);
        }
    }
    for(int i = 1 ; i <= n ; i ++)
    {
        if(s[i] < s[k] && v[i] > min_v || s[i] > s[k] && v[i] <max_v)
        {
            ans++;
        }
    }
    cout<<ans;
}