P3406 海底高铁（前缀和+差分+坑点）

P3406 海底高铁https://www.luogu.com.cn/problem/P3406https://www.luogu.com.cn/problem/P3406

分析

本题是一个典型的前缀和和差分的例题，题目难度不大，把题目样例画一下就能有大致思路
1. 题目的一个考察点在每条路段都有2种方案，办卡或者不办卡，那么就要考虑最便宜的情况

什么时候最便宜？
 办卡的总消费C1 = 卡费 + 优惠价 * 经过次数
 不办卡的总费用C2 = 原价 * 经过次数
如果C1 < C2，那么就选择办卡，这里的的未知数只有一个——>经过次数

 2. 现在问题就来到了怎么求经过次数，如果你选择一个个点的去模拟，那么这个题你就会TLE，所以这里我们采用差分的方法来实现

何为差分：
我们可以发现，每次一个区间内的数同时加上一个数，它们的差值是不会变的。这也是差分的思想。

用数组一个b，b[i]=a[i]-a[i-1],即b[i]表示第i项较前一项（第i-1项）的增量。 需要注意的是，b[1]=a[1]。 （同样的，我们可以优化到不使用a数组，不过需要记录前一项的值）

当[x,y]区间内的数都加上z时，a[x]+=z(第x项比其前一项的增量增加z）， a[y+1]-=z(第y+1项比其前一项的增量减少z，因为第y项变大了！)。 这样我们就很轻松地维护了区间内的变化。

显然，由于b[i]都是第i项较前一项的增量，我们将b数组累加就可以求得指定项的数。即，a[i]=b[1]+b[2]+...+b[i]（这里用前缀和来加）
 

这里你就可以发现：用差分来增加每段的经过次数，只要给头和尾做操作，而不用差分的话就要给区间内的所有数+1，时间消耗就大大增加了

AC CODE

#include <iostream>
using namespace std;
const int maxn = 100000 + 5;
typedef long long ll;
struct rail
{
    ll ori_pri, card, sal_pri;
} rail[maxn];

ll sum[maxn], co[maxn], ans;
ll n, m, a[maxn], cnt[maxn], b[maxn];

int main()
{
    cin >> n >> m;
    for (int i = 0; i < m; i++)
        cin >> a[i];
    for (int i = 1; i < n; i++)
        cin >> rail[i].ori_pri >> rail[i].sal_pri >> rail[i].card;
    for (int i = 0; i < m - 1; i++)
    {
        int l, r;
        if (a[i] > a[i + 1])
        {
            l = a[i + 1];
            r = a[i];
        }
        else
        {
            l = a[i];
            r = a[i + 1];
        }
        //维护cnt[l,r)，cnt[l,r)中每个值都+1，表示经过次数
        b[l] += 1;
        b[r] -= 1;
    }
    b[0] = cnt[0] = 0;
    for (int i = 1; i < n; i++)
        cnt[i] = b[i] + cnt[i - 1];

    for (int i = 1; i < n; i++)
    {
        ll c1 = rail[i].card + rail[i].sal_pri * cnt[i];
        ll c2 = rail[i].ori_pri * cnt[i];
        if (c1 < c2)
        {
            ans += rail[i].card;
            co[i] = rail[i].sal_pri;
        }
        else
            co[i] = rail[i].ori_pri;
    }

    for (int i = 1; i < n; i++)
        sum[i] = sum[i - 1] + co[i];

    for (int i = 0; i < m - 1; i++)
    {
        int next = a[i + 1];
        ans += a[i] > a[i + 1] ? (ll)sum[a[i] - 1] - (ll)sum[a[i + 1] - 1] : (ll)sum[a[i + 1] - 1] - (ll)sum[a[i] - 1];
    }

    cout << ans;
    return 0;
}

坑点

自己给数据都开了long long却依然通不过，后来发现了问题所在

我在最后求和的时候采用了如下比较的方法

for (int i = 0; i < m - 1; i++)
    {
        int next = a[i + 1];
        ans += a[i] > a[i + 1] ? sum[a[i] - 1] - sum[a[i + 1] - 1] : sum[a[i + 1] - 1] - sum[a[i] - 1];
    }

不知道是不是在这种比较大小里

ans = a > b ? sum[a]-sum[b] : sum[b]-sum[a] 

对于返回给ans的(sum[a] - sum[b]）并没有用long long返回所以就出错了

如果改成

for (int i = 0; i < m - 1; i++)
    {
        int next = a[i + 1];
        if (a[i] > a[i + 1])
            ans += sum[a[i] - 1] - sum[a[i + 1] - 1];
        else
            ans += sum[a[i + 1] - 1] - sum[a[i] - 1];
    }

就对了 或者在前面加ll也能对

for (int i = 0; i < m - 1; i++)
    {
        int next = a[i + 1];
        ans += a[i] > a[i + 1] ? (ll)sum[a[i] - 1] - (ll)sum[a[i + 1] - 1] : (ll)sum[a[i + 1] - 1] - (ll)sum[a[i] - 1];
    }

同理 max和min函数应该也要这样做