Submission #35597377

---

Source Code Expand

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
#define rep(i,a,n) for (ll i=a;i<(ll)n;i++)

#define SEG_ROOT 1,0,n-1
#define SEG_L (o<<1)
#define SEG_R (o<<1|1)
#define mid (l+r)/2
#define SEG_L_CHILD SEG_L,l,mid
#define SEG_R_CHILD SEG_R,mid+1,r

ll read(){ll r;scanf("%lld",&r);return r;}
using A3=array<ll,3>;
using A33=array<A3,3>;

struct Node{
  A3 a={0,0,0};
  A3 lazy={0,1,2}; // 未向下传递的(本层已经作用
  A33 c={{{{0,0,0}},{{0,0,0}},{{0,0,0}}}};
}seg[400010]; // seg[0] empty node
int a[100010];

template<class T>
T merge(const T&v0,const T&v1){
  T v;
  rep(i,0,3)v.a[i]=v0.a[i]+v1.a[i];
  rep(i,0,3)rep(j,0,3)v.c[i][j]=v0.c[i][j]+v1.c[i][j];
  rep(i,0,3)rep(j,0,3)if(i!=j) v.c[i][j] += v0.a[i]*v1.a[j];
  return v;
}

Node build(int o,int l,int r){
  if(l==r){
    seg[o].a[a[l]]=1;
    return seg[o];
  }
  return seg[o]=merge(build(SEG_L_CHILD),build(SEG_R_CHILD));
}

void zotfn(int o,const A3&zot){ // zero one two
  Node newo=seg[0];
  rep(i,0,3)newo.a[zot[i]]+=seg[o].a[i];
  rep(i,0,3)newo.lazy[i] = zot[seg[o].lazy[i]];
  rep(i,0,3)rep(j,0,3)newo.c[zot[i]][zot[j]]+=seg[o].c[i][j];
  seg[o]=newo;
}

void down(int o){
  zotfn(SEG_L,seg[o].lazy);
  zotfn(SEG_R,seg[o].lazy);
  seg[o].lazy={0,1,2};
}

Node query(int o,int l,int r,int ql,int qr){
  if(ql<=l&&r<=qr)return seg[o];
  down(o);
  Node resl = ql<=mid?query(SEG_L_CHILD,ql,qr):seg[0];
  Node resr = qr> mid?query(SEG_R_CHILD,ql,qr):seg[0];
  return merge(resl,resr);
}

void update(int o,int l,int r,int ql,int qr,const A3& zot){
  if(ql<=l&&r<=qr) return zotfn(o,zot);
  down(o);
  if(ql<=mid) update(SEG_L_CHILD,ql,qr,zot);
  if(qr> mid) update(SEG_R_CHILD,ql,qr,zot);
  seg[o]=merge(seg[SEG_L],seg[SEG_R]);
}

int main(){
  int n=read();
  int q=read();
  rep(i,0,n)a[i]=read();
  build(SEG_ROOT);
  while(q--){
    int op=read();
    int l=read()-1;
    int r=read()-1;
    if(op==1){
      auto x=query(SEG_ROOT,l,r).c;
      printf("%lld\n",x[1][0]+x[2][0]+x[2][1]);
    }else{
      A3 zot;
      rep(i,0,3)zot[i]=read();
      update(SEG_ROOT,l,r,zot);
    }
  }
  return 0;
}

Submission Info

Compile Error

./Main.cpp: In function ‘ll read()’:
./Main.cpp:13:21: warning: ignoring return value of ‘int scanf(const char*, ...)’, declared with attribute warn_unused_result [-Wunused-result]
   13 | ll read(){ll r;scanf("%lld",&r);return r;}
      |                ~~~~~^~~~~~~~~~~

Judge Result