PTA L3 - 区区区间3 (30 分)
题意
输入样例
5 2 3
1 5 3 2 4
2 1 5
1 1 4
输出样例
11
样例解释
进行操作2 1 5
后,序列变为1 3 2 5 4
所以操作1 1 4
的答案为1+3+2+5=11
附加输入
7 16 4
1 5 3 2 4 6 7
2 2 6
1 1 1 1 2 2 1 3 3 1 4 4 1 5 5 1 6 6 1 7 7
3 3 5
1 1 1 1 2 2 1 3 3 1 4 4 1 5 5 1 6 6 1 7 7
附加输出
1 3 2 4 5 6 7
1 3 5 2 4 6 7
思路
①
首先能够注意到,整组数据中的 x x x值是定的,所以可以根据序列中每个数与 x x x之间的大小关系先分成两组
※以附加样例为例,初始数列为 { 1 , 5 , 3 , 2 , 4 , 6 , 7 } \{1,5,3,2,4,6,7\} { 1,5,3,2,4,6,7}, x = 4 x=4 x=4,分组后得到
a [ i ] ≤ x a[i]\le x a[i]≤x: { a r } = { 1 , 3 , 2 , 4 } \{ar\}=\{1,3,2,4\} { ar}={ 1,3,2,4}
a [ i ] > x a[i]\gt x a[i]>x: { b r } = { 5 , 6 , 7 } \{br\}=\{5,6,7\} { br}={ 5,6,7}
②
又注意到,不论是操作 2 2 2还是操作 3 3 3,都没有打乱上述分组的元素之间的顺序
所以不论怎么操作,每组中的数字在原序列中的下标一定是个递增序列
※以附加样例为例,初始时两组数字的下标序列为(按顺序对应数字)
a [ i ] ≤ x a[i]\le x a[i]≤x: { a r p o s } = { 1 , 3 , 4 , 5 } \{ar_{pos}\}=\{1,3,4,5\} { arpos}={ 1,3,4,5}
a [ i ] > x a[i]\gt x a[i]>x: { b r p o s } = { 2 , 6 , 7 } \{br_{pos}\}=\{2,6,7\} { brpos}={ 2,6,7}
在进行了操作2 2 6
后,下标序列变成
a [ i ] ≤ x a[i]\le x a[i]≤x: { a r p o s } = { 1 , 2 , 3 , 4 } \{ar_{pos}\}=\{1,2,3,4\} { arpos}={ 1,2,3,4}
a [ i ] > x a[i]\gt x a[i]>x: { b r p o s } = { 5 , 6 , 7 } \{br_{pos}\}=\{5,6,7\} { brpos}={ 5,6,7}
再进行操作3 3 5
后,下标序列变成
a [ i ] ≤ x a[i]\le x a[i]≤x: { a r p o s } = { 1 , 2 , 4 , 5 } \{ar_{pos}\}=\{1,2,4,5\} { arpos}={ 1,2,4,5}
a [ i ] > x a[i]\gt x a[i]>x: { b r p o s } = { 3 , 6 , 7 } \{br_{pos}\}=\{3,6,7\} { brpos}={ 3,6,7}
③
于是可以发现,操作 2 2 2在下标序列中的修改可以理解成以下几步操作:
1、选出 { a r p o s } \{ar_{pos}\} { arpos}中范围在 [ l , r ] [l,r] [l,r]间数字的区间 [ r 1 l , r 1 r ] [r1l,r1r] [r1l,r1r]
2、选出 { b r p o s } \{br_{pos}\}