CodeForces - 981 G Magic multisets (선분 트 리 + set 유지보수 구간 정보)

G. Magic multisets
time limit per test
4 seconds
memory limit per test
256 megabytes
input
standard input
output
standard output
In the School of Magic in Dirtpolis a lot of interesting objects are studied on Computer Science lessons.
Consider, for example, the magic multiset. If you try to add an integer to it that is already presented in the multiset, each element in the multiset duplicates. For example, if you try to add the integer 22 to the multiset {1,2,3,3}{1,2,3,3}, you will get {1,1,2,2,3,3,3,3}{1,1,2,2,3,3,3,3}.
If you try to add an integer that is not presented in the multiset, it is simply added to it. For example, if you try to add the integer 44 to the multiset {1,2,3,3}{1,2,3,3}, you will get {1,2,3,3,4}{1,2,3,3,4}.
Also consider an array of nn initially empty magic multisets, enumerated from 11 to nn.
You are to answer qq queries of the form "add an integer xx to all multisets with indices l,l+1,…,rl,l+1,…,r" and "compute the sum of sizes of multisets with indices l,l+1,…,rl,l+1,…,r". The answers for the second type queries can be large, so print the answers modulo 998244353998244353.
Input
The first line contains two integers nn and qq (1≤n,q≤2⋅1051≤n,q≤2⋅105) — the number of magic multisets in the array and the number of queries, respectively.
The next qq lines describe queries, one per line. Each line starts with an integer tt (1≤t≤21≤t≤2) — the type of the query. If tt equals 11, it is followed by three integers ll, rr, xx (1≤l≤r≤n1≤l≤r≤n, 1≤x≤n1≤x≤n) meaning that you should add xx to all multisets with indices from ll to rrinclusive. If tt equals 22, it is followed by two integers ll, rr (1≤l≤r≤n1≤l≤r≤n) meaning that you should compute the sum of sizes of all multisets with indices from ll to rr inclusive.
Output
For each query of the second type print the sum of sizes of multisets on the given segment.
The answers can be large, so print them modulo 998244353998244353.
Examples
input
Copy
4 4
1 1 2 1
1 1 2 2
1 1 4 1
2 1 4

output
Copy
10

input
Copy
3 7
1 1 1 3
1 1 1 3
1 1 1 2
1 1 1 1
2 1 1
1 1 1 2
2 1 1

output
Copy
4
8

Note
In the first example after the first two queries the multisets are equal to [{1,2},{1,2},{},{}][{1,2},{1,2},{},{}], after the third query they are equal to [{1,1,2,2},{1,1,2,2},{1},{1}][{1,1,2,2},{1,1,2,2},{1},{1}].
In the second example the first multiset evolves as follows:
{}→{3}→{3,3}→{2,3,3}→{1,2,3,3}→{1,1,2,2,3,3,3,3}{}→{3}→{3,3}→{2,3,3}→{1,2,3,3}→{1,1,2,2,3,3,3,3}.
 
 
문제 풀이 방향: 먼저 N 개의 집합 크기 를 하나의 선분 트 리 로 유지 하고 이 선분 트 리 는 구간 곱셈, 구간 덧셈, 구간 구 화 를 지원 합 니 다.관건 은 업데이트 작업 이다.언제 구간 이 * 2 또는 + 1 이 고 어떤 구간 이 * 2 또는 + 1 이 어야 합 니까?
그러면 우 리 는 모든 집합 안에 어떤 숫자 가 있 는 지 지 켜 야 한다. 만약 우리 가 주석 나무 로 발견 하 는 것 이 좀 어렵다 면.그럼 폭력 을 생각해.
매번 구간 을 갱신 할 때마다 우 리 는 모든 집합 을 폭력 적 으로 조회 하여 그 수가 존재 하 는 지 확인 하 는데, 이렇게 하 는 것 은 분명히 바람 직 하지 않다.
그러면 우 리 는 다른 방식 으로 생각 하 는 것 도 좋 습 니 다. 우 리 는 하나의 집합 배열 S [i] 로 i 번 째 숫자 가 어떤 구간 에 나 타 났 는 지 유지 합 니 다.
우리 가 업데이트 작업 이 있 을 때 우 리 는 S [w] 집합 을 직접 조회 한 다음 에 이 집합 안의 모든 구간 이 조회 한 [L, R] 과 교차 하 는 것 이 있 는 지 판단 하고 교차 하 는 부분 은 * 2 로 교차 하지 않 는 부분 은 + 1 로 한다. 이런 방법 은 매우 좋 은 것 같 지만 최 악의 복잡 한 것 은 매우 크다.그래서 우 리 는 어떤 구간 이 교차 하 는 지 효율적으로 조회 해 야 한다.그래서 S [w] 집합 은 구간 을 저장 할 수 없고 구간 의 점 을 저장 해 야 합 니 다. 그러면 우 리 는 set 자체 의 lower 를 사용 할 수 있 습 니 다.bound 작업, 효율 적 인 업데이트 구간 과 집합.코드 조작 을 구체 적 으로 보다.
set 유지보수 구간 을 배 워 보 세 요.
 
 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
using namespace std;
typedef unsigned long long ull;
typedef long long ll;
const int MAXN = 200005;
const ll MOD = 998244353;

ll lazyAdd[MAXN<<2];
ll sum[MAXN<<2];
ll lazyMul[MAXN<<2];
int N,Q;

void build(int l,int r,int rt){
    lazyMul[rt]=1;
    if(l==r)
        return;
    int m=(l+r)/2;
    build(l,m,rt<<1);
    build(m+1,r,rt<<1|1);
}

void pushup(int rt){
    sum[rt]=(sum[rt<<1]+sum[rt<<1|1])%MOD;
}

void pushdown(int rt,int ln,int rn){
    if(lazyMul[rt]!=1){
        sum[rt<<1]=(sum[rt<<1]%MOD*lazyMul[rt]%MOD)%MOD;
        lazyMul[rt<<1]=(lazyMul[rt<<1]%MOD*lazyMul[rt]%MOD)%MOD;
        lazyAdd[rt<<1]=(lazyAdd[rt<<1]%MOD*lazyMul[rt]%MOD)%MOD;
        sum[rt<<1|1]=(sum[rt<<1|1]%MOD*lazyMul[rt]%MOD)%MOD;
        lazyMul[rt<<1|1]=(lazyMul[rt<<1|1]%MOD*lazyMul[rt]%MOD)%MOD;
        lazyAdd[rt<<1|1]=(lazyAdd[rt<<1|1]%MOD*lazyMul[rt]%MOD)%MOD;
        lazyMul[rt]=1;
    }
    if(lazyAdd[rt]){
        sum[rt<<1]=(sum[rt<<1]+lazyAdd[rt]*ln%MOD)%MOD;
        lazyAdd[rt<<1]=(lazyAdd[rt<<1]+lazyAdd[rt]%MOD)%MOD;
        sum[rt<<1|1]=(sum[rt<<1|1]+lazyAdd[rt]*rn%MOD)%MOD;
        lazyAdd[rt<<1|1]=(lazyAdd[rt<<1|1]+lazyAdd[rt])%MOD;
        lazyAdd[rt]=0;
    }
}

void updateAdd(int L,int R, ll C,int l,int r,int rt){
    if(L<=l&&r<=R){
        lazyAdd[rt]=(lazyAdd[rt]+C)%MOD;
        sum[rt]=(sum[rt]+C*(r-l+1)%MOD)%MOD;
        return;
    }
    int m=(l+r)/2;
    pushdown(rt,m-l+1,r-m);
    if(L<=m)
        updateAdd(L,R,C,l,m,rt<<1);
    if(R>m)
        updateAdd(L,R,C,m+1,r,rt<<1|1);
    pushup(rt);
}

void updateMul(int L,int R,ll C,int l,int r,int rt){
    if(L<=l&&r<=R){
        lazyMul[rt]=(lazyMul[rt]*C)%MOD;
        lazyAdd[rt]=(lazyAdd[rt]*C)%MOD;
        sum[rt]=(sum[rt]*C)%MOD;
        return;
    }
    int m=(l+r)/2;
    pushdown(rt,m-l+1,r-m);
    if(L<=m)
        updateMul(L,R,C,l,m,rt<<1);
    if(R>m)
        updateMul(L,R,C,m+1,r,rt<<1|1);
    pushup(rt);
}

ll query(int L,int R,int l,int r,int rt){
    if(L<=l&&r<=R)
        return sum[rt]%MOD;
    int m=(l+r)/2;
    pushdown(rt,m-l+1,r-m);
    ll ans=0;
    if(L<=m)
        ans=(ans+query(L,R,l,m,rt<<1))%MOD;
    if(R>m)
        ans=(ans+query(L,R,m+1,r,rt<<1|1))%MOD;
    return ans%MOD;
}

set > S[MAXN];

/***      ***/
void update(int l,int r,int w){
    auto &s=S[w];
    pair node1=make_pair(l,0);
    pair node2=make_pair(r,1);
    
    while(l<=r){
        auto it=s.lower_bound(make_pair(l,0));
        if(it==s.end()){
            updateAdd(l,r,1,1,N,1);
            break;
        }
        if(it->second==1){
            updateMul(l,min(it->first,r),2,1,N,1);
            l=min(it->first,r)+1;
            if(r>=it->first)
                s.erase(it);
        }
        else{
            if(l<=min(it->first-1,r))//        l    
                updateAdd(l,min(it->first-1,r),1,1,N,1);
            l=min(it->first-1,r)+1;
            if(r>=it->first)
                s.erase(it);
        }
    }
    
    auto it=s.lower_bound(node1);
    auto it2=s.lower_bound(node1);
    if(it==s.begin()||(--it)->second)
        s.insert(node1);
    if(it2==s.end()||it2->second==0)
        s.insert(node2);
}


int main()
{
    
    scanf("%d%d",&N,&Q);
    build(1,N,1);
    int op,l,r,w;
    for(int i=0;i

 
 
 
 
 
 
 
 
 
 

좋은 웹페이지 즐겨찾기