結果
| 問題 |
No.879 Range Mod 2 Query
|
| コンテスト | |
| ユーザー |
tubuann
|
| 提出日時 | 2019-09-03 12:35:46 |
| 言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
| 結果 |
AC
|
| 実行時間 | 377 ms / 3,000 ms |
| コード長 | 4,851 bytes |
| コンパイル時間 | 2,715 ms |
| コンパイル使用メモリ | 210,848 KB |
| 最終ジャッジ日時 | 2025-01-07 16:22:04 |
|
ジャッジサーバーID (参考情報) |
judge2 / judge4 |
(要ログイン)
| ファイルパターン | 結果 |
|---|---|
| sample | AC * 1 |
| other | AC * 21 |
ソースコード
#include<bits/stdc++.h>
using namespace std;
typedef unsigned long long int ull;
typedef long long int ll;
typedef pair<ll,ll> pll;
typedef long double D;
typedef complex<D> P;
#define F first
#define S second
const ll E=1e18+7;
const ll MOD=1000000007;
template<typename T,typename U>istream & operator >> (istream &i,pair<T,U> &A){i>>A.F>>A.S; return i;}
template<typename T>istream & operator >> (istream &i,vector<T> &A){for(auto &I:A){i>>I;} return i;}
template<typename T,typename U>ostream & operator << (ostream &o,const pair<T,U> &A){o<<A.F<<" "<<A.S; return o;}
template<typename T>ostream & operator << (ostream &o,const vector<T> &A){ll i=A.size(); for(auto &I:A){o<<I<<(--i?" ":"");} return o;}
template<typename T>vector<T> & cset(vector<T> &A,T e=T()){for(auto &I:A){I=e;} return A;}
#ifndef SEGMENT_TREE
#define SEGMENT_TREE
template<typename Data,typename Mono>
class SegTree{
private:
typedef function<void(Data&,Data&)> MergeData;
typedef function<void(Mono&,Mono&)> MergeMono;
typedef function<void(Data&,Mono&)> Culc;
typedef function<bool(Data&,Mono&,int&,int&)> UpdateQuery; //data,mono,left,right true::Data,Monoに作用する false::子に降りる
int size,high;
vector<int> left,right;
vector<Data> dv;
vector<Mono> mv;
Data de;
Mono me;
MergeData df;
MergeMono mf;
Culc cf;
inline bool nx(int &idx){
while(idx&1){idx>>=1; if(idx){reculc(idx);}}
return idx?idx|=1:idx;
}
inline void reculc(int idx){
dv[idx]=dv[idx<<1];
df(dv[idx],dv[(idx<<1)|1]);
//cf(dv[idx],mv[idx]);
}
inline void down(int idx){
cf(dv[idx<<1],mv[idx]);
mf(mv[idx<<1],mv[idx]);
cf(dv[(idx<<1)|1],mv[idx]);
mf(mv[(idx<<1)|1],mv[idx]);
mv[idx]=me;
}
public:
SegTree(Data de,Mono me,MergeData df,MergeMono mf,Culc cf,int n=0):de(de),me(me),df(df),mf(mf),cf(cf){init(n);}
void init(int n){
size=1;
high=0;
while(size<n){size<<=1; high++;}
dv.assign(size<<1,de);
mv.assign(size<<1,me);
left.resize(size<<1);
right.resize(size<<1);
for(int i=0;i<size;i++){left[size|i]=i; right[size|i]=i+1;}
for(int i=size-1;i>0;i--){left[i]=left[i<<1]; right[i]=right[(i<<1)|1];}
}
void build(const vector<Data> &ary){
int n=ary.size();
init(n);
for(int i=0;i<n;i++){dv[size|i]=ary[i];}
for(int i=size-1;i>0;i--){dv[i]=dv[i<<1]; df(dv[i],dv[(i<<1)|1]);}
}
//[lf,rg)
Data query(int lf,int rg){
Data ret=de;
int idx=1;
bool j=true;
while(j){
if(right[idx]<=lf || rg<=left[idx]){j&=nx(idx);}
else if(lf<=left[idx] && right[idx]<=rg){df(ret,dv[idx]); j&=nx(idx);}
else{down(idx); idx<<=1;}
}
return ret;
}
void update(int lf,int rg,Mono m){
int idx=1;
bool j=true;
while(j){
if(right[idx]<=lf || rg<=left[idx]){j&=nx(idx);}
else if(lf<=left[idx] && right[idx]<=rg){
cf(dv[idx],m); mf(mv[idx],m); j&=nx(idx);
}
else{down(idx); idx<<=1;}
}
}
void update(int lf,int rg,UpdateQuery br){
int idx=1;
bool j=true;
while(j){
if(right[idx]<=lf || rg<=left[idx]){j&=nx(idx);}
else if(lf<=left[idx] && right[idx]<=rg){
if(br(dv[idx],mv[idx],left[idx],right[idx])){j&=nx(idx);}
else{down(idx); idx<<=1;}
}
else{down(idx); idx<<=1;}
}
}
void DEBUG(){cout<<dv<<endl;}
};
#endif /*SEGMENT_TREE*/
int main(){
cin.tie(0);
ios::sync_with_stdio(false);
typedef pair<pll,pll> P;//odd(num,sum),even(num,sum)
typedef pll P2;//0::none,1::+1mod2,2::mod2
P de={{0,0},{0,0}};
P2 me={0,0};
auto df=[](P &p1,P &p2){
p1.F.F+=p2.F.F;
p1.F.S+=p2.F.S;
p1.S.F+=p2.S.F;
p1.S.S+=p2.S.S;
};
auto mf=[](P2 &p1,P2 &p2){
if(p2.F==0){
p1.S+=p2.S;
}
else{
p1.F+=p1.S+p2.F;
p1.F&=1;
if(p1.F==0){p1.F+=2;}
p1.S=p2.S;
}
};
auto cf=[](P &p,P2 &p2){
if(p2.F==1){
swap(p.F,p.S);
}
if(p2.F!=0){
p.F.S=p.F.F;
p.S.S=0;
}
if(p2.S&1){
swap(p.F,p.S);
}
p.F.S+=p.F.F*p2.S;
p.S.S+=p.S.F*p2.S;
};
ll n,q;
cin>>n>>q;
vector<ll> A(n);
cin>>A;
vector<P> arr(n,de);
for(int i=0;i<n;i++){
if(A[i]&1){arr[i]={{1,A[i]},{0,0}};}
else{arr[i]={{0,0},{1,A[i]}};}
}
SegTree<P,P2> Tr(de,me,df,mf,cf);
Tr.build(arr);
for(int i=0;i<q;i++){
int c,l,r,x;
cin>>c>>l>>r;
l--; r--;
if(c==1){
Tr.update(l,r+1,{2,0});
}
else if(c==2){
cin>>x;
Tr.update(l,r+1,{0,x});
}
else{
auto I=Tr.query(l,r+1);
cout<<I.F.S+I.S.S<<endl;
}
}
return 0;
}
tubuann