沙拉酱非常喜欢数字序列。这正是他要弄一个关于构造序列的算法的原因。
沙拉酱拿了一张白纸。然后他开始用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
记录每一次操作的行为,记录当前行为的位置,不断二分位置,向前找。
做这道题的时候想起了http://acm.hdu.edu.cn/showproblem.php?pid=5587,感觉这两道题本质上的思想差不多,后者需要找规律,然后不断向前翻。这道题是自己标记断点,二分位置,向前翻。
代码:
#pragma warning(disable:4996)
#include <iostream>
#include <algorithm>
#include <cmath>
#include <vector>
#include <string>
#include <cstring>
#include <queue>
#include <map>
using namespace std;
typedef long long ll;
const int maxn = 1e5 + 5;
int n;
ll pos[maxn];
struct ope
{
int val;
int oper;//当前命令
int x;//添加的数字
int li;//前缀的长度
int ci;//复制的次数
}op[maxn];
void input()
{
int i, temp;
scanf("%d", &n);
for (i = 1; i <= n; i++)
{
scanf("%d", &op[i].oper);
if (op[i].oper == 1)
{
scanf("%d", &temp);
pos[i] = pos[i - 1] + 1;
op[i].val = temp;
}
else
{
scanf("%d%d", &op[i].li, &op[i].ci);
pos[i] = pos[i - 1] + op[i].li*op[i].ci;
}
}
}
void solve()
{
int i, k;
ll ans;
scanf("%d", &k);
for (i = 1; i <= k; i++)
{
scanf("%lld", &ans);
while (true)
{
int p = lower_bound(pos + 1, pos + n, ans) - pos;
if (op[p].oper == 1)
{
printf("%d ", op[p].val);
break;
}
else
{
ans = ans - pos[p - 1];
ans = ans%op[p].li;
if (ans == 0)
{
ans = op[p].li;
}
}
}
}
}
int main()
{
//freopen("i.txt", "r", stdin);
//freopen("o.txt", "w", stdout);
input();
solve();
return 0;
}