1560. Elementary Symmetric Functions

这是一道很有意思的题目。
首先整体上看过去，有单点修改和区间查询的操作，所以基本可以确定用线段树来做。
然后仔细观察一下，可以发现一个非常有意思的性质。
我们考虑将两个区间合并，$s[0/1][5]$表示两个区间的S函数。
如果在每一个区间都选择两个数，那么贡献值就是$$(\sum _{l_1<=x_1<y_1<=r_1}a[x_1]\ast a[y_1])\ast (\sum _{l_2<=x_2<y_2<=r_2}a[x_2]\ast a[y_2])=s[0][2]\ast s[1][2]$$
根据这个性质，所以我们直接给线段树每一个节点上建立一个数组$c[5]$，然后利用卷积合并即可。
查询的时候一个个合并就行了。

code

#include<bits/stdc++.h>
#define ll long long
#define gc getchar
#define pc putchar
#define pii pair<int,int>
#define mp make_pair
#define fi first
#define se second
#define rep(i,x,y) for(int i=x;i<=y;i++)
#define dwn(i,x,y) for(int i=x;i>=y;i--)
using namespace std;
template<typename T>inline void qr(T &x){
  x=0;int f=0;char s=gc();
  while(s<'0'||'9'<s)f|=s=='-',s=gc();
  while('0'<=s&&s<='9')x=x*10+s-48,s=gc();
  x=f?-x:x;
}
int cc=0,buf[31];
template<typename T>inline void qw(T x){
  if(x<0)putchar('-'),x=-x;
  do{buf[++cc]=int(x%10);x/=10;}while(x);
  while(cc)pc(buf[cc--]+'0');pc(' ');
}
const int N=8e4+10;
int n,m,mod,a[N];
int ad(int x,int y){return (x+y)>=mod?x+y-mod:x+y;}
int mu(int x,int y){return 1ll*x*y%mod;}
int de(int x,int y){return (x-y)<0?x-y+mod:x-y;}
struct node{
  int c[5];
  node(){memset(c,0,sizeof(c));}
};
node operator*(node x,node y){
  node ans;
  rep(i,0,4)rep(j,0,i)ans.c[i]=ad(ans.c[i],mu(x.c[j],y.c[i-j]));
  return ans;
}
#define lc p<<1
#define rc p<<1|1
#define mid (t[p].l+t[p].r)/2
struct tree{int l,r;node c;}t[N<<2];
void build(int p,int l,int r){
  t[p].l=l,t[p].r=r;
  if(l<r){
    build(lc,l,mid),build(rc,mid+1,r);
    t[p].c=t[lc].c*t[rc].c;
  }
  else t[p].c.c[0]=1,t[p].c.c[1]=a[l];
}
void change(int p,int x,int k){
  if(t[p].l==t[p].r){
    t[p].c.c[1]=ad(t[p].c.c[1],k);
    return;
  }
  if(x<=mid)change(lc,x,k);
  else change(rc,x,k);
  t[p].c=t[lc].c*t[rc].c;
}
node ans;
void query(int p,int l,int r){
  if(l<=t[p].l&&t[p].r<=r){
    ans=ans*t[p].c;return;
  }
  if(l<=mid)query(lc,l,r);
  if(mid<r)query(rc,l,r);
}
int main(){
  qr(n),qr(m),qr(mod);
  rep(i,1,n){qr(a[i]);a[i]=(a[i]%mod+mod)%mod;}
  build(1,1,n);
  while(m--){
    char ss[2];int x,y;
    scanf("%s",ss);qr(x),qr(y);
    if(ss[0]=='I'){
      y=(y%mod+mod)%mod;
      change(1,x,y);
    }
    else{
      memset(ans.c,0,sizeof(ans.c));ans.c[0]=1;
      query(1,x,y);
      int k;qr(k);
      rep(i,0,k)qw(ans.c[i]);
      puts("");
    }
  }
  return 0;
}