洛谷P3406 海底高铁(前缀和&差分）Java题解

海底高铁

题目传送门：P3406 海底高铁 - 洛谷 | 计算机科学教育新生态 (luogu.com.cn)

题目描述

该铁路经过 $N$ 个城市，每个城市都有一个站。不过，由于各个城市之间不能协调好，于是乘车每经过两个相邻的城市之间（方向不限），必须单独购买这一小段的车票。第 $i$ 段铁路连接了城市 $i$ 和城市 $i+1(1\le i<N)$。如果搭乘的比较远，需要购买多张车票。第 $i$ 段铁路购买纸质单程票需要 $A_i$ 博艾元。

虽然一些事情没有协调好，各段铁路公司也为了方便乘客，推出了 IC 卡。对于第 $i$ 段铁路，需要花 $C_i$ 博艾元的工本费购买一张 IC 卡，然后乘坐这段铁路一次就只要扣 $B_i(B_i<A_i)$ 元。IC 卡可以提前购买，有钱就可以从网上买得到，而不需要亲自去对应的城市购买。工本费不能退，也不能购买车票。每张卡都可以充值任意数额。对于第 $i$ 段铁路的 IC 卡，无法乘坐别的铁路的车。

Uim 现在需要出差，要去 $M$ 个城市，从城市 $P_1$ 出发分别按照 $P_1,P_2,P_3,\cdots ,P_M$ 的顺序访问各个城市，可能会多次访问一个城市，且相邻访问的城市位置不一定相邻，而且不会是同一个城市。

现在他希望知道，出差结束后，至少会花掉多少的钱，包括购买纸质车票、买卡和充值的总费用。

输入格式

第一行两个整数，$N,M$。

接下来一行，$M$ 个数字，表示 $P_i$。

接下来 $N-1$ 行，表示第 $i$ 段铁路的 $A_i,B_i,C_i$。

输出格式

一个整数，表示最少花费

样例 #1

样例输入 #1

9 10
3 1 4 1 5 9 2 6 5 3
200 100 50
300 299 100
500 200 500
345 234 123
100 50 100
600 100 1
450 400 80
2 1 10

样例输出 #1

6394

提示

$2$ 到 $3$ 以及 $8$ 到 $9$ 买票，其余买卡。

对于 $30\%$ 数据 $M=2$。

对于另外 $30\%$ 数据 $N\le 1000，M\le 1000$。

对于 $100\%$ 的数据 $M,N\le 10^5，A_i,B_i,C_i\le 10^5$。

解题思路：

涉及知识：前缀和&差分

3 1 4 1 5 9 2 6 5 3中的前两位数3 1 表示从P3城市到P1城市，会途经P2城市，同理1 4表示从P1到P4会途经 P2和P3。本题的重点即是得出每段路要走多少次，以此来判断是买卡的费用少还是直接买票的费用少。

每段路要走的次数可以用前缀和&差分计算

测试案例描述如下表所示：

根据前缀和的性质，要想在区间的一个数组d[1]-d[3]之间加上1，则在d[1]++之后还需要对d[3]--，加了又减回来才能不变。

求出差分数组后，再求得前缀和数组即是每段路要经过的次数，最后再比较买卡费用少还是买票费用小，贪心求取即可。

注意事项： 前缀和数组、每段路的费用以及最终费用可能很大，应使用long型，避免暴开。

java实现代码如下：

package base.qianzuihe_chafen;
import java.io.*;
public class P3406_海底高铁 {
    public static void main(String[] args) throws IOException{
        BufferedReader bu = new BufferedReader(new InputStreamReader(System.in));
        StreamTokenizer in = new StreamTokenizer(bu);
        PrintWriter out = new PrintWriter(new OutputStreamWriter(System.out));
        in.nextToken();
        int N = (int)in.nval;
        in.nextToken();
        int M = (int)in.nval;
        int n = 100004;
        int [] p = new int[M+1];
        long [] c = new long[n];//差分数组
        long [] sum = new long[n];//前缀和数组
        int[][] a = new int[n][3];//费用数组
        for (int i = 1; i < M+1; i++) {
            in.nextToken();
            p[i] = (int)in.nval;
        }
        for (int i = 1; i <N ; i++) {
            for (int j = 0; j < 3; j++) {
                in.nextToken();
                a[i][j] = (int)in.nval;
            }
        }
        //计算差分数组
        for (int i = 1; i <M ; i++) {
            int min = Math.min(p[i],p[i+1]);
            int max = Math.max(p[i],p[i+1]);
            c[min]++;
            c[max]--;
        }
        //计算前缀和数组
        for (int i = 1; i <n-1 ; i++) {
            sum[i] = c[i]+sum[i-1];
        }
        //计算最小费用
        long res = 0;
        for (int i = 1; i <N ; i++) {
            long x = Math.min(sum[i]*a[i][0],sum[i]*a[i][1]+a[i][2]);
            res += x;
        }
        if(M==0 || M==1) res = 0;
        out.println(res);
        out.flush();
    }
}

如果本文对你有帮助，可以点个赞再走噢！