ATcoder-Replace Digits【线段树】

正题

题目链接:https://atcoder.jp/contests/abl

---

题目大意

$n$个数字开始全是$1$，要求支持

1. 修改一个区间为一个数字（是$1\sim 9$的数）
2. 求所有数字串起来$\%998244353$

---

解题思路

其实就是第$i$个数字乘上$10^i$，我们考虑如何用线段树维护这个东西，我们可以维护一个$s_x={\sum }_{i=1}^{x}10^i$，然后区间修改时我们就可以用$s$来计算这个位置的值。

时间复杂度$O(n\log n)$

---

$code$

#include<cstdio>
#include<cstring>
#include<algorithm>
#define ll long long
using namespace std;
const ll N=2e5+10,XJQ=998244353;
ll n,q,s[N];
ll lazy[N*4],t[N*4];
void Downdata(ll x,ll l,ll m,ll r){
	if(!lazy[x])return;
	ll val=lazy[x];
	lazy[x*2]=lazy[x*2+1]=lazy[x];
	t[x*2]=(s[m]-s[l-1]+XJQ)%XJQ*val%XJQ;
	t[x*2+1]=(s[r]-s[m]+XJQ)%XJQ*val%XJQ;
	lazy[x]=0;return;
}
void Change(ll x,ll L,ll R,ll l,ll r,ll val){
	if(L==l&&R==r){
		lazy[x]=val;
		t[x]=(s[R]-s[L-1]+XJQ)%XJQ*val%XJQ;
		return;
	}
	ll mid=(L+R)>>1;
	Downdata(x,L,mid,R);
	if(r<=mid)Change(x*2,L,mid,l,r,val);
	else if(l>mid)Change(x*2+1,mid+1,R,l,r,val);
	else Change(x*2,L,mid,l,mid,val),Change(x*2+1,mid+1,R,mid+1,r,val);
	t[x]=(t[x*2]+t[x*2+1])%XJQ;
}
int main()
{
	scanf("%lld%lld",&n,&q);
	s[1]=1;
	for(ll i=2;i<=n;i++)
		s[i]=s[i-1]*10%XJQ;
	for(ll i=1;i<=n;i++){
		s[i]=(s[i]+s[i-1])%XJQ;
		Change(1,1,n,i,i,1);
	}
	while(q--){
		ll l,r,d;
		scanf("%lld%lld%lld",&l,&r,&d);
		l=n-l+1;r=n-r+1;swap(l,r);
		Change(1,1,n,l,r,d);
		printf("%lld\n",t[1]);
	}
	return 0;
}