沙拉酱非常喜欢数字序列。这正是他要弄一个关于构造序列的算法的原因。
沙拉酱拿了一张白纸。然后他开始用m个步骤来制作一个序列。每一步他要么向这个序列的末尾添加一个数字,要么拿这个序列的开头l个数字,然后在末尾添加c次。对于第二种操作,一般的,如果当前序列是 a1,a2,...,an ,那么经过操作之后序列将变成 a1,a2,...,an[,a1,a2,...,al] (方括号里面的内容会重复c次)。
一天过去了,沙拉酱也完成了他的序列。现在他想知道某个位置是什么数字。
Input
单组测试数据。 第一行包含一个整数m (1 ≤ m ≤ 10^5),表示构造序列的步骤数目。 接下来m行包含每一个步骤的信息。第一个数字是类型(1或2)。类型1表示在序列后面加一个数字,这种情况下后面会跟一个整数xi (1 ≤ xi ≤ 10^5),表示被加在后面的数字。类型2表示复制一段长度为 li 前缀然后接到后面 ci 次,这种情况下后面会跟两个整数 li, ci(1 ≤ li ≤ 10^5, 1 ≤ ci ≤ 10^4),li 是前缀的长度,ci是复制的次数。输入中保证li不会大于当前序列的长度。 接下来一行包含一个整数n (1 ≤ n ≤ 10^5),表示查询的数量。接下来一行中包含n个正整数,每一个整数表示要查询的位置。题目保证这些数字大小不会超过序列的长度。序列的下标从1开始。
Output
对于每一个查询,输出对应查询位置的数字。两个查询之间用空格分开。具体格式看样例。
Input示例
6 1 1 1 2 2 2 1 1 3 2 5 2 1 4 16 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16
Output示例
1 2 1 2 3 1 2 1 2 3 1 2 1 2 3 4
算出每次操作 前数列中已经有k个数,那么该操作就是在k+1个位置上添加数字,对于每个操作保存k+1,用于二分操作.
对于询问第m个位置上的数是多少,二分找到第m个位置是在第i操作上产生的,若该操作的操作类型是1直接输出答案,否则将将m -=m - 该操作的(k+1) + 1, m %= l, 那么就减小所查询的数的位置,循环操作
#include <bits/stdc++.h>
#define MOD 1000000007
#define maxn 100005
#define INF 1e18
using namespace std;
typedef long long ll;
struct Node{
int t, x, l, c;
ll d;
}node[maxn];
int m;
int Search(ll v){
int l = 0, r = m;
while(l < r){
int mid = (l + r) >> 1;
if(node[mid].d > v)
r = mid;
else
l = mid + 1;
}
return l - 1;
}
int main(){
// freopen("in.txt", "r", stdin);
ll ans = 0;
scanf("%d", &m);
for(int i = 0; i < m; i++){
scanf("%d", &node[i].t);
node[i].d = ans + 1;
if(node[i].t == 1){
scanf("%d", &node[i].x);
ans++;
}
else{
scanf("%d%d", &node[i].l, &node[i].c);
ans += node[i].l * node[i].c;
}
}
ll v;
int n, first = 0;
scanf("%d", &n);
while(n--){
scanf("%I64d", &v);
int h = Search(v);
while(node[h].t != 1){
v = v - node[h].d + 1;
v %= node[h].l;
if(v == 0)
v = node[h].l;
h = Search(v);
}
if(first)
putchar(' ');
first = 1;
printf("%d", node[h].x);
}
puts("");
return 0;
}