oj.uz

답안 #284574

# 제출 시각 아이디 문제 언어 결과 실행 시간 메모리
284574 2020-08-27T16:38:38 Z limabeans Sterilizing Spray (JOI15_sterilizing) C++17
25 / 100
 203 ms 7672 KB 
#include <bits/stdc++.h>
using namespace std;

template<typename T>
void out(T x) { cout << x << endl; exit(0); }
#define watch(x) cout << (#x) << " is " << (x) << endl





using ll = long long;

const ll mod = 1e9+7;
const int maxn = 1e5 + 5;

template<typename T> struct segtree {
    
    T merge(T x, T y) {
	x += y;
	return x;
    }
    int n;
    vector<ll> t;
    void init(int n) {
	n += 10;
	this->n = n;
	t.resize(n*4);
    }

    void upd(int v, int tl, int tr, int i, T val) {
	if (tl == tr) {
	    t[v] = val;
	} else {
	    int tm = (tl+tr)/2;
	    if (i<=tm) {
		upd(2*v,tl,tm,i,val);
	    } else {
		upd(2*v+1,tm+1,tr,i,val);
	    }
	    t[v] = merge(t[2*v], t[2*v+1]);
	}
    }

    T qry(int v, int tl, int tr, int l, int r) {
	if (l>tr || r<tl) {
	    return 0;
	}
	if (l <= tl && tr <= r) {
	    return t[v];
	}
	int tm = (tl+tr)/2;
	return merge(qry(2*v,tl,tm,l,r), qry(2*v+1,tm+1,tr,l,r));
    }
};

int n, q, k;
int a[maxn];
int S[maxn], T[maxn], U[maxn];

void brute(int type, int t, int u, ll& res) {
    if (type==1) {
	int idx = t;
	int val = u;
	a[idx] = val;
    }
    if (type==2) {
	int l = t;
	int r = u;
	for (int i=l; i<=r; i++) {
	    a[i] /= k;
	}
    }
    if (type==3) {
	res = 0;
	int l = t;
	int r = u;
	for (int i=l; i<=r; i++) {
	    res += a[i];
	}
    }
}


void solveTask1() {
    for (int i=0; i<q; i++) {
	int s = S[i];
	int t = T[i];
	int u = U[i];
	ll bans=-1; brute(s,t,u,bans);
	if (s==3) cout<<bans<<"\n";
    }
}

// Task2 on oj.uz
void solveTask2() {
    segtree<ll> tree;
    tree.init(n+10);
    for (int i=1; i<=n; i++) {
	tree.upd(1,1,n,i,a[i]);
    }
    for (int i=0; i<q; i++) {
	if (S[i]==1) {
	    tree.upd(1,1,n,T[i],U[i]);
	}
	if (S[i]==3) {
	    cout<<tree.qry(1,1,n,T[i],U[i])<<"\n";
	}
    }
}

//////////////////////////////////////////////////////////////////
ll t[4*maxn];
ll mark[4*maxn];

const int NONE = -1;

void push(int v, int tl, int tr) {
    if (mark[v]!=NONE) {
	int tm=(tl+tr)/2;
	t[2*v]=1ll*(tm-tl+1)*mark[v];
	t[2*v+1]=1ll*(tr-tm)*mark[v];
	mark[v*2]=mark[2*v+1]=mark[v];
	mark[v]=NONE;
    }
}

ll queryRange(int v, int tl, int tr, int l, int r) {
    if (l>r) return 0;
    if (l==tl && tr==r) {
	return t[v];
    }
    push(v,tl,tr);
    int tm=(tl+tr)/2;
    return queryRange(2*v,tl,tm,l,min(tm,r))
	+  queryRange(2*v+1,tm+1,tr,max(tm+1,l),r);
}

void setRange(int v, int tl, int tr, int l, int r, int val) {
    if (l>r) return;
    if (l==tl && tr==r) {
	t[v]=1ll*(tr-tl+1)*val;
	mark[v]=val;
	return;
    }
    push(v,tl,tr);
    int tm=(tl+tr)/2;
    setRange(2*v,tl,tm,l,min(tm,r),val);
    setRange(2*v+1,tm+1,tr,max(tm+1,l),r,val);
    t[v]=t[2*v]+t[2*v+1];
}

//////////////////////////////////////////////////////////////////

void solveTask3() {
    //cout<<"Task3"<<endl;
    for (int i=1; i<=n; i++) {
	setRange(1,1,n,i,i,a[i]);
    }

    for (int i=0; i<q; i++) {
	// ll bans=-1;
	// brute(S[i],T[i],U[i],bans);
	
	if (S[i]==1) {
	    setRange(1,1,n,T[i],T[i],U[i]);
	}
	if (S[i]==2) {
	    if (k>1) {
		setRange(1,1,n,T[i],U[i],0);
	    }
	}
	if (S[i]==3) {
	    ll res = queryRange(1,1,n,T[i],U[i]);
	    //assert(res==bans);
	    cout<<res<<endl;
	}
    }
}


int main() {
    ios_base::sync_with_stdio(false); cin.tie(0);  cout.tie(0);

    cin>>n>>q>>k;
    bool isTask3 = true;
    for (int i=1; i<=n; i++) {
	cin>>a[i];
	if (!(0<=a[i] && a[i]<=1)) isTask3=false;
    }
    for (int i=0; i<q; i++) {
	cin>>S[i]>>T[i]>>U[i];
	if (S[i]==1) {
	    if (!(0<=U[i] && U[i]<=1)) isTask3=false;
	}
    }

    if (k==1) {
	solveTask2();
	exit(0);
    }

    if (isTask3) {
	solveTask3();
	exit(0);
    }

    if (n<=3000 && q<=3000) {
	solveTask1();
	exit(0);
    }

    
    assert(false);
    
    
    return 0;
}

Subtask #1
5.0 / 5.0

# 결과 실행 시간 메모리 Grader output
1 Correct 1 ms 384 KB Output is correct
2 Correct 2 ms 384 KB Output is correct
3 Correct 1 ms 384 KB Output is correct
4 Correct 4 ms 512 KB Output is correct
5 Correct 6 ms 512 KB Output is correct
6 Correct 7 ms 512 KB Output is correct
7 Correct 6 ms 512 KB Output is correct
8 Correct 6 ms 512 KB Output is correct
9 Correct 7 ms 512 KB Output is correct
10 Correct 7 ms 512 KB Output is correct
11 Correct 7 ms 512 KB Output is correct
12 Correct 7 ms 512 KB Output is correct

Subtask #2
10.0 / 10.0

# 결과 실행 시간 메모리 Grader output
1 Correct 59 ms 5000 KB Output is correct
2 Correct 53 ms 4132 KB Output is correct
3 Correct 50 ms 5372 KB Output is correct
4 Correct 64 ms 6668 KB Output is correct
5 Correct 78 ms 6456 KB Output is correct
6 Correct 75 ms 6648 KB Output is correct
7 Correct 78 ms 6520 KB Output is correct
8 Correct 78 ms 6512 KB Output is correct
9 Correct 73 ms 6520 KB Output is correct
10 Correct 73 ms 6520 KB Output is correct
11 Correct 75 ms 6672 KB Output is correct
12 Correct 72 ms 6520 KB Output is correct

Subtask #3
10.0 / 10.0

# 결과 실행 시간 메모리 Grader output
1 Correct 62 ms 1656 KB Output is correct
2 Correct 36 ms 3072 KB Output is correct
3 Correct 50 ms 3328 KB Output is correct
4 Correct 144 ms 3832 KB Output is correct
5 Correct 203 ms 7672 KB Output is correct
6 Correct 190 ms 7672 KB Output is correct
7 Correct 69 ms 6588 KB Output is correct
8 Correct 197 ms 7416 KB Output is correct
9 Correct 186 ms 7288 KB Output is correct
10 Correct 187 ms 7476 KB Output is correct
11 Correct 173 ms 7480 KB Output is correct
12 Correct 183 ms 7440 KB Output is correct

Subtask #4
0 / 75.0

# 결과 실행 시간 메모리 Grader output
1 Runtime error 27 ms 3712 KB Execution killed with signal 11
2 Halted 0 ms 0 KB -