제출 #1273698

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
1273698		nthvn	Sterilizing Spray (JOI15_sterilizing)	C++20	0 / 100	107 ms	22448 KiB

sterilizing

#include "bits/stdc++.h"
using namespace std;

#define fi first
#define se second
#define pii pair<int,int> 
#define all(x) x.begin(), x.end()
#define sz(x) (int)x.size()
#define pb push_back
#define ll long long
const int N = 1e5+5;
const int LG = 20;
int n,q,k, a[N];

struct Node{
	int dt[LG], lz;
	void combine(Node &a, Node &b){
		for(int i=0;i<LG;i++) dt[i] = a.dt[i]+ b.dt[i];
	}
	ll toNum(){
		ll ans =0;
		for(int i=LG-1;i>=0;i--){
			ans = ans*k + dt[i];
		}
		return ans;
	}
}tr[4*N];

void app_lz(Node &a, int lz){
	a.lz+= lz;
	if(lz>=LG) {
		memset(a.dt,0,sizeof(a.dt));
		return;
	}
	memmove(a.dt, a.dt+lz, (LG-lz)*sizeof(lz));
}

void down(int id){
	if(!tr[id].lz) return;
	int t= tr[id].lz;
	app_lz(tr[2*id], t);
	app_lz(tr[2*id+1],t);
	tr[id].lz =0;
}

void encode(int dt[], int x){
	for(int i=0;i<LG;i++){
		dt[i] = x%k;
		x/=k;
	}
}

void build(int id, int l, int r){
	if(l==r){
		encode(tr[id].dt, a[l]);
		return;
	}
	int mid=(l+r)>>1;
	build(2*id,l,mid);
	build(2*id+1,mid+1,r);
	tr[id].combine(tr[2*id], tr[2*id+1]);
}
void upd1(int id, int l, int r, int p, int v){
	if(l>p||r<p) return;
	if(l==r){
		encode(tr[id].dt, v);
		return;
	}
	int mid=(l+r)>>1;
	down(id);
	upd1(2*id,l,mid,p,v);
	upd1(2*id+1,mid+1,r,p,v);
	tr[id].combine(tr[2*id], tr[2*id+1]);
}

void upd2(int id, int l, int r, int t_l ,int t_r){
	if(l>t_r||r<t_l) return;
	if(t_l<=l&&r<=t_r){
		app_lz(tr[id], 1);
		return;
	}
	int mid=(l+r)>>1;
	down(id);
	upd2(2*id,l,mid,t_l,t_r);
	upd2(2*id+1,mid+1,r,t_l,t_r);
	tr[id].combine(tr[2*id], tr[2*id+1]);
}

ll get(int id, int l, int r, int t_l ,int t_r){
	if(l>t_r||r<t_l) return 0;
	if(t_l<=l&&r<=t_r) return tr[id].toNum();
	int mid=(l+r)>>1;
	down(id);
	return get(2*id,l,mid,t_l,t_r)+get(2*id+1,mid+1,r,t_l,t_r);
}

signed main(){
	ios_base::sync_with_stdio(NULL);
	cin.tie(NULL);
	if(fopen("TASK.INP", "r")){
		freopen("TASK.INP", "r", stdin);
		freopen("TASK.OUT", "w", stdout);
	}
	cin>>n>>q>>k;
	for(int i=1;i<=n;i++) cin>>a[i];
	build(1,1,n);
	while(q--){
		int t,u,v; cin>>t>>u>>v;
		if(t==1){
			upd1(1,1,n,u,v);
		}
		else if(t==2){
			upd2(1,1,n,u,v);
		}
		else cout<<get(1,1,n,u,v)<<"\n";
	}
}

컴파일 시 표준 에러 (stderr) 메시지

sterilizing.cpp: In function 'int main()':
sterilizing.cpp:101:24: warning: ignoring return value of 'FILE* freopen(const char*, const char*, FILE*)' declared with attribute 'warn_unused_result' [-Wunused-result]
  101 |                 freopen("TASK.INP", "r", stdin);
      |                 ~~~~~~~^~~~~~~~~~~~~~~~~~~~~~~~
sterilizing.cpp:102:24: warning: ignoring return value of 'FILE* freopen(const char*, const char*, FILE*)' declared with attribute 'warn_unused_result' [-Wunused-result]
  102 |                 freopen("TASK.OUT", "w", stdout);
      |                 ~~~~~~~^~~~~~~~~~~~~~~~~~~~~~~~~

• Subtask 1: 0 / 5
• Subtask 2: 0 / 10
• Subtask 3: 0 / 10
• Subtask 4: 0 / 75