小可求和问题

目描述

现在有两个长度为 nn的数组，有一个询问：op,l,r。

1. op=1 表示求数组 a中第 l个元素到第 r 个元素的和。
2. op=2表示求数组 b中第 l 个元素到第 r 个元素的和。

小可需要回答这个答案。

输入格式

• 第一行一个整数 n
• 第二行 n 个整数，表示 a 数组的元素
• 第三行 n 个整数，表示 b 数组的元素
• 第四行三个整数 op,l,r

输出格式

• 一个整数，表示答案。

样例

输入数据 1

3
1 2 3
4 5 6
1 1 2

输出数据 1

3

输入数据 2

4
1 2 3 4
5 6 7 8
2 3 4

输出数据 2

15

数据范围

100% 的数据： n≤1000,1≤ai,bi≤106,l≤r

代码

#include<bits/stdc++.h>
using namespace std;
int main(){
    int n;
    cin>>n;
    int a[n],b[n];
    for(int i=0;i<n;i++){
        cin>>a[i];
    }
    for(int i=0;i<n;i++){
        cin>>b[i];
    }
    int op,l,r,ans=0;
    cin>>op>>l>>r;
    if(op==1){
        for(int i=l-1;i<r;i++){
            ans=ans+a[i];
        }
    }else{
         for(int i=l-1;i<r;i++){
            ans=ans+b[i];
        }
    }
    cout<<ans;
}