[语言月赛 202307] 署前街少年
题目背景
时光无法缝补那块破碎的黑板
虚荣的少年与署前街越来越远
——赵雷,《署前街少年》
题目描述
某 E 得到了一个长度为 2 N 2N 2N 的数列 a 1 , a 2 , a 3 , … a 2 N a_1, a_2, a_3, \dots a_{2N} a1,a2,a3,…a2N,数列的第 i i i 个数为 a i a_i ai。
奇变偶不变,符号看象限。这是三角函数诱导公式的重要口诀。某 E 同样想对数列实施这样的变换,具体来说:
- 对于 a i a_i ai,若 i m o d 2 = 0 i \bmod 2=0 imod2=0,则称 a i a_i ai 为偶位数;若 i m o d 2 = 1 i \bmod 2 = 1 imod2=1,则称 a i a_i ai 为奇位数。
- 对于 a i a_i ai,记 i m o d k = p i \bmod k = p imodk=p,则称 a i a_i ai 为第 p p p 象限数,其中 k k k 为给定的参数。
奇变偶不变,符号看象限。某 E 将遵循以下的规则对数列进行变换:
- 若 a i a_i ai 为偶位数,则 a i a_i ai 不变。
- 若 a i a_i ai 为奇位数,设 a i a_i ai 为第 p p p 象限数,则 a i a_i ai 变为所有第 p p p 象限 数 ( a i ) (a_i) (ai) 的和对 i i i 取模的值。
请注意以上变换不会影响「所有第 p p p 象限 数 ( a i ) (a_i) (ai) 的和」这一数值。
某 E 想知道,变换后的数列是什么样的。
输入格式
输入共两行。
输入的第一行为两个整数 N , k N,k N,k。
输入的第二行为 2 N 2N 2N 个整数,第 i i i 个代表 a i a_i ai。
输出格式
输出一行 2 N 2N 2N 个整数,第 i i i 个代表变换后的 a i a_i ai。
样例 #1
样例输入 #1
10 4
5 2 0 4 7 6 2 7 1 3 5 7 9 45 3 6 12 36 78 1
样例输出 #1
0 2 1 4 4 6 4 7 7 3 0 7 8 45 13 6 0 36 12 1
提示说明
- 对于 40 % 40\% 40% 的测试数据, 1 ≤ N ≤ 1000 1 \le N \le 1000 1≤N≤1000, 2 ≤ k ≤ 10 2 \le k \le 10 2≤k≤10。
- 对于 100 % 100\% 100% 的测试数据, 1 ≤ N ≤ 5 × 1 0 5 1 \le N \le 5 \times 10^5 1≤N≤5×105, 2 ≤ k ≤ 1 0 4 2 \le k \le 10^4 2≤k≤104, 0 ≤ a i ≤ 1 0 6 0 \le a_i \le 10^6 0≤ai≤106 。
代码内容
// #include <iostream>
// #include <algorithm>
// #include <cstring>
// #include <stack>//栈
// #include <deque>//队列
// #include <queue>//堆/优先队列
// #include <map>//映射
// #include <unordered_map>//哈希表
// #include <vector>//容器,存数组的数,表数组的长度
#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const ll N=1e6+10;
ll a[N];
ll s[N];
int main()
{
ll n,k;
cin>>n>>k;
for(ll i=1;i<=2*n;i++)
{
ll p;
cin>>a[i];
p=i%k;
s[p]+=a[i];//相同象限的a[i]累加
}
for(ll i=1;i<=2*n;i++)
{
ll p;
//奇位数
if(i%2==1)
{
p=i%k;
//a[i]:所有第 p 象限数的和对 i 取模的值
a[i]=s[p]%i;
}
cout<<a[i]<<" ";
}
return 0;
}