2018洛谷8月月赛第一题_U28036 Nagisa loves Tomoya

U28036 Nagisa loves Tomoya

题目描述：

可爱的古河渚得到了一个序列。她定义一次操作为将所有的a_i变为a_i+a_{(i mod n+1)}

然后她会向你进行Q次询问,每次问你x,y意义为在第x次操作后位置y的元素是多少。

这个问题太简单了,我们的Nagisa轻易地就解决了这个问题。请问您能解决这个问题吗？

输入格式：

一个整数n,表示序列的长度。

接下来一行n个整数,表示这个序列。

接下来一行一个整数Q,表示询问次数

接下来Q行,每行两个整数x,y意义如题所示。

输出格式：

共Q行,每行一个整数表示询问的答案,请注意对998244353取膜。

题解：

找到翻x次每个数字被统计的次数的规律，即杨辉三角。

当k>n的时候，就拐回来，即第一项加上最后一项。

代码：

const
  mo=998244353;
var
  n,q:longint;
  a,f:array [0..1000001] of int64;
  x,y,ans,b:array [0..10001] of int64;
function min(o,p:longint):longint;
begin
  if o<p then exit(o);
  exit(p);
end;

procedure init;
var
  i:longint;
begin
  fillchar(ans,sizeof(ans),0);
  fillchar(f,sizeof(f),0);
  readln(n);
  for i:=1 to n do
    read(a[i]);
  a[0]:=a[n];
  readln(q);
  for i:=1 to q do
    begin
      readln(x[i],y[i]);
      b[i]:=i;
    end;
end;

procedure qsort(l,r:longint);
var
  i,j,t,mid:longint;
begin
  if l>r then exit;
  i:=l; j:=r;
  mid:=x[(l+r) div 2];
  repeat
    while x[i]<mid do inc(i);
    while x[j]>mid do dec(j);
    if i<=j then
      begin
        t:=x[i]; x[i]:=x[j]; x[j]:=t;
        t:=y[i]; y[i]:=y[j]; y[j]:=t;
        t:=b[i]; b[i]:=b[j]; b[j]:=t;
        inc(i); dec(j);
      end;
  until i>j;
  qsort(i,r);
  qsort(l,j);
end;

procedure main;
var
  i,j,k,t:longint;
begin
  f[0]:=1;
  i:=1; k:=0;
  while i<=q do
    begin
      while x[i]=k do
        begin
          for j:=0 to min(k,n-1) do
            ans[b[i]]:=(ans[b[i]]+f[j]*a[(y[i]+j) mod n]) mod mo;
          inc(i);
        end;
      while k<x[i] do
        begin
          t:=f[n-1];
          for j:=min(n,k) downto 1 do
            f[j]:=(f[j]+f[j-1]) mod mo;
          f[0]:=(f[0]+t) mod mo;
          inc(k); f[k]:=1;
        end;
    end;
end;

procedure print;
var
  i:longint;
begin
  for i:=1 to q do
    writeln(ans[i]);
end;

begin
  init;
  qsort(1,q);
  main;
  print;
end.