问题描述
南将军手下有N个士兵,分别编号1到N,这些士兵的杀敌数都是已知的。
小工是南将军手下的军师,南将军经常想知道第m号到第n号士兵的总杀敌数,请你帮助小工来回答南将军吧。
南将军的某次询问之后士兵i可能又杀敌q人,之后南将军再询问的时候,需要考虑到新增的杀敌数。
输入格式
多组测试数据,以EOF结尾;
每组第一行是两个整数N,M,其中N表示士兵的个数(1<N<1000000),M表示指令的条数。(1<M<100000)
随后的一行是N个整数,ai表示第i号士兵杀敌数目。(0<=ai<=100)
随后的M行每行是一条指令,这条指令包含了一个字符串和两个整数,首先是一个字符串,如果是字符串QUERY则表示南将军进行了查询操
作,后面的两个整数m,n,表示查询的起始与终止士兵编号;如果是字符串ADD则后面跟的两个整数I,A(1<=I<=N,1<=A<=100),表示第I个士兵新增杀敌数为A.
输出格式
对于每次查询,输出一个整数R表示第m号士兵到第n号士兵的总杀敌数,每组输出占一行
样例输入
5 6
1 2 3 4 5
QUERY 1 3
ADD 1 2
QUERY 1 3
ADD 2 3
QUERY 1 2
QUERY 1 5
样例输出
6
8
8
20
思路:
根据题目的意思创建数组存放士兵的杀敌数,然后根据命令执行相应操作。应该注意的是它一组测评数据含有多组测试数据,用Java写可以加上while (sc.hasNext())来时所有数据都读取执行,最开始我没注意,未加while (sc.hasNext()),它只执行了一组数据,导致错误。
我的代码可能比别人多,但我也是按照我这个初学者的思路来写的,对于刚接触算法的朋友可能更容易理解
完整代码:
import java.util.Scanner;
public class Main {
static int[] soldior;
public static void main(String[] args) {
Scanner sc = new Scanner(System.in);
while (sc.hasNext()){ // 因为有多组数据,所有应执行到所有数据都读取
int n = sc.nextInt(); // 输入n和m
int m = sc.nextInt();
sc.nextLine(); // 当上一行是数字,下一行为字符串时,要用sc.nextLine()消除回车的影响
String[] nums = sc.nextLine().trim().split(" "); // 接收士兵的杀敌数
soldior = new int[n+1];
for (int i=1;i<=n;i++) // 给各士兵杀敌数赋值,为了方便可以让i从1开始而不是0
soldior[i] = Integer.parseInt(nums[i-1]);
String[] orders = new String[m]; // 命令的行数
for (int i=0;i<m;i++){
orders[i] = sc.nextLine(); // 保存命令
actOrder(orders[i]); // 执行命令
}
}
}
private static void actOrder(String order){ // 根据不同的命令执行不同的操作
String[] temp = order.split(" ");
int i = Integer.parseInt(temp[1]); // 转换为整型,方便操作
int j = Integer.parseInt(temp[2]);
switch (temp[0]){ // 根据命令执行不同方法
case "QUERY":
query(i,j);
break;
case "ADD":
add(i,j);
}
}
private static void query(int a, int b){ // 查询操作
int sum = 0;
for (int i=a;i<=b;i++)
sum += soldior[i];
System.out.println(sum);
}
private static void add(int a, int b){ // 增加操作
soldior[a] += b;
}
}