沙拉酱非常喜欢数字序列。这正是他要弄一个关于构造序列的算法的原因。
沙拉酱拿了一张白纸。然后他开始用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
思路:
记录每次输入的值和增加的长度。用二分法找到每个位置的值的原始输入。
#include <iostream>
using namespace std;
typedef long long int ll;
const int MAXN = 100005;
int a[MAXN][3];
ll num[MAXN];
int main(void)
{
int m, n, y;
cin >> m;
for (int i = 1; i <= m; i++)
{
cin >> a[i][0];
if (a[i][0] == 1)
{
cin >> a[i][1];
num[i] = num[i-1]+1;
}
else
{
cin >> a[i][1] >> a[i][2];
num[i] = num[i-1] + a[i][1] * a[i][2];
}
}
cin >> n;
ll pos;
for (int i = 1; i <= n; i++)
{
cin >> pos;
while (true)
{
y = lower_bound(num, num + m + 1, pos) - num;
if (a[y][0]==1)
{
cout << a[y][1] << " ";
break;
}
else
{
pos = (pos - num[y - 1] - 1) % a[y][1] + 1;
}
}
}
cout << endl;
return 0;
}