贝茜放慢脚步(二路归并)

题目链接

题意：给定n个减速信息，计算走完一千米所需的时间；减速信息分为关于时间的减速点和关于距离的减速点

思路：先将时间减速点和路程减速点分别排序，然后不断比较当前时间减速点a[i]和路程减速点b[j]哪个先发生，设当前的路程s，时间t，速度v，则距a[i]发生还差时间的为a[i]-t，距b[j]发生还差的时间为(b[j]-s)/v，直至处理完所有减速点，输出时间

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int maxx = 1e5 + 10;
const int inf=0x3f3f3f3f;
int n;
vector<int>a,b;//关于时间的减速点，关于距离的减速点
void solve()
{
    cin>>n;
    char op[2];
    int x;
    while(n--)
    {
        scanf("%s%d",op,&x);
        if(*op=='T') a.push_back(x);
        else b.push_back(x);
    }
    b.push_back(1000);//加入终点
    sort(a.begin(),a.end());
    sort(b.begin(),b.end());
    double s=0,t=0,v=1;//v为速度的倒数
    int i=0,j=0;
    while(i<a.size()||j<b.size())//二路归并
    {
        if(j==b.size()||(i<a.size()&&a[i]-t<(b[j]-s)*v))//距离减速点空了||(距离减速点没空，时间减速点没空，时间减速点先发生)
        {
            s+=(a[i]-t)/v;//注意更新顺序
            t=a[i];
            v++;
            i++;//指针后移
        }
        else
        {
            t+=(b[j]-s)*v;
            s=b[j];
            v++;
            j++;
        }
    }
    printf("%.0f",t);
}
int main()
{
    int _t=1;
    //scanf("%d", &_t);
    while (_t--)
    {
        solve();
    }
    //system("pause");
}