[语言月赛202208] 数组调整
题目描述
给出一个长度为 n n n 的数组,第 i i i 个数为 a i a_i ai。
为了调整这个数组,需要将第 k k k 个数改变为 − a k -a_k −ak。
请你求出调整后的数组中所有数的和。
输入格式
输入共两行。
输入的第一行为两个整数 n , k n,k n,k。
输入的第二行为 n n n 个整数,第 i i i 个为 a i a_i ai。
输出格式
输出一行一个整数,代表调整后的数组中所有数的和。
样例 #1
样例输入 #1
10 7
1 2 3 4 5 6 7 8 9 10
样例输出 #1
41
样例 #2
样例输入 #2
5 2
1 999 4 3 2
样例输出 #2
-989
样例 #3
样例输入 #3
10 2
1 0 1 1 1 1 1 1 1 1
样例输出 #3
9
提示说明
输入输出样例 1 解释
调整后的数组为 1 2 3 4 5 6 -7 8 9 10
,和为 41。
输入输出样例 2 解释
调整后的数组为 1 -999 4 3 2
,和为 -989。
数据规模与约定
对于 5 % 5\% 5% 的数据,为输入输出样例 1。
对于另外 40 % 40\% 40% 的数据,需要调整位置上的数 a k = 0 a_k=0 ak=0。
对于另外 20 % 20\% 20% 的数据, k = 1 k = 1 k=1。
对于 100 % 100\% 100% 的数据, 1 ≤ k ≤ n , n ≤ 1 0 6 , 0 ≤ a i ≤ 1 0 9 1 \le k \le n, n \le 10^6,0 \le a_i \le 10^9 1≤k≤n,n≤106,0≤ai≤109。
代码内容
// #include <iostream>
// #include <algorithm>
// #include <cstring>
// #include <sstream>//整型转字符串
// #include <stack>//栈
// #include <deque>//堆/优先队列
// #include <queue>//队列
// #include <map>//映射
// #include <unordered_map>//哈希表
// #include <vector>//容器,存数组的数,表数组的长度
#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
int main()
{
ll n,k;
cin>>n>>k;
ll sum=0;
for(ll i=1;i<=n;i++)
{
ll x;
cin>>x;
if(i==k) sum-=x;
else sum+=x;
}
cout<<sum<<endl;
return 0;
}