题目描述
You have N integers A1, A2, ... , AN. You are asked to write a program to receive and execute two kinds of instructions:
- C a b means performing Ai = (Ai2 mod 2018) for all Ai such that a ≤ i ≤ b.
- Q a b means query the sum of Aa, Aa+1, ..., Ab. Note that the sum is not taken modulo 2018.
输入描述:
The first line of the input is T(1≤ T ≤ 20), which stands for the number of test cases you need to solve.
The first line of each test case contains N (1 ≤ N ≤ 50000).The second line contains N numbers, the initial values of A1, A2, ..., An. 0 ≤ Ai < 2018. The third line contains the number of operations Q (0 ≤ Q ≤ 50000). The following Q lines represents an operation having the format "C a b" or "Q a b", which has been described above. 1 ≤ a ≤ b ≤ N.
输出描述:
For each test case, print a line "Case #t:" (without quotes, t means the index of the test case) at the beginning.
You need to answer all Q commands in order. One answer in a line.
分析
刚开始打表算错了周期。。。又坑队友了
每个元素平方最大周期为6,且6是其他所有周期的公倍数。我们可以在每个节点维护一个大小为6的数组,同时维护一个代表从数组中取哪个元素的指针。因为有的数在进入循环节前需要经过几次修改,打表发现进入循环节前的修改次数都不超过5,所以可以在前五次暴力更新节点,之后才开始利用线段树的lazy标记。
代码
#include <iostream>
#include <cstdio>
#include <cstring>
#include <vector>
#include <stack>
#include <algorithm>
using namespace std;
typedef long long ll;
const int mod=2018;
const int maxn=50050;
int cnt[maxn*4],sum[maxn*4][7],lazy[maxn*4],p[maxn*4];
void build(int l,int r,int id) {
cnt[id]=p[id]=lazy[id]=0;
if(l==r) {
scanf("%d", &sum[id][0]);
}
else {
int mid=(l+r)>>1;
build(l,mid,id<<1);
build(mid+1,r,id<<1|1);
sum[id][0]=sum[id<<1][0]+sum[id<<1|1][0];
}
}
void pushUp(int id) {
cnt[id]=min(cnt[id<<1],cnt[id<<1|1]);
if(cnt[id]>=5) {
p[id]=0;
for (int i = 0; i < 6; ++i)
sum[id][i]=sum[id<<1][(p[id<<1]+i)%6]+sum[id<<1|1][(p[id<<1|1]+i)%6];
}
else sum[id][0]=sum[id<<1][p[id<<1]]+sum[id<<1|1][p[id<<1|1]];
}
void pushDown(int id) {
p[id<<1]=(p[id<<1]+lazy[id])%6,lazy[id<<1]+=lazy[id];
p[id<<1|1]=(p[id<<1|1]+lazy[id])%6,lazy[id<<1|1]+=lazy[id];
lazy[id]=0;
}
void modify(int x,int y,int l,int r,int id) {
if(l>r||l>y||r<x) return;
if(l==r) {
cnt[id]++;
if(cnt[id]<5) {
sum[id][0]=sum[id][0]*sum[id][0]%mod;
}
else if(cnt[id]==5) {
p[id]=0;
sum[id][0]=sum[id][0]*sum[id][0]%mod;
for (int i = 1; i < 6; ++i)
{
sum[id][i]=sum[id][i-1]*sum[id][i-1]%mod;
}
}
else {
p[id]=(p[id]+1)%6;
}
return;
}
if(lazy[id]) pushDown(id);
if(x<=l&&y>=r) {
int mid=(l+r)>>1;
if(cnt[id]<5) {
modify(x,y,l,mid,id<<1);
modify(x,y,mid+1,r,id<<1|1);
pushUp(id);
}
else {
lazy[id]++;
p[id]=(p[id]+1)%6;
}
return;
}
int mid=(l+r)>>1;
if(x<=mid) modify(x,y,l,mid,id<<1);
if(y>mid) modify(x,y,mid+1,r,id<<1|1);
pushUp(id);
}
int query(int x,int y,int l,int r,int id) {
if(l!=r && lazy[id]) pushDown(id);
if(l>=x&&r<=y) {
return sum[id][p[id]%6];
}
int mid=(l+r)>>1;
int ans=0;
if(x<=mid) ans+=query(x,y,l,mid,id<<1);
if(y>mid) ans+=query(x,y,mid+1,r,id<<1|1);
return ans;
}
int main(int argc, char const *argv[])
{
int t,Case=0;
scanf("%d", &t);
while(t--)
{
int n;
scanf("%d", &n);
build(1,n,1);
int q;
scanf("%d", &q);
printf("Case #%d:\n", ++Case);
while(q--)
{
char str[3];
int l,r;
scanf("%s%d%d", str,&l,&r);
if(str[0]=='Q') {
printf("%d\n", query(l,r,1,n,1));
}
else {
modify(l,r,1,n,1);
}
}
}
return 0;
}