[luogu] P1370 Charlie的云笔记序列 线性dp+离散化

前言

传送门 :

题意

给定$n,a[]$,对于 a [ l , r ] = { a l . . . . a r } a[l,r]=\{a_l....a_r\} a[l,r]={al​....ar​}

定义$F_{l,r}=a[l,r]中不同子序列的数量$

特别的 空串也算一种 (子序列不一定要连续

求${\sum }_{l=1}^n{\sum }_{r=1}^{n}F[l,r]$

思路

这种题我们考虑$dp$

状态表示 :
$f[i]$ 表示${\sum }_{j=1}^i{F}_{ij}$

状态计算 :
$f[i]=f[i-1]+f[i-1]+2$

因为我们加入了$a[i]$,因此$f[i]$又多了$f[i]$种,同时 a [ i ] , { } a[i],\{\} a[i],{}也需要记录到答案中

又因为不能出现相同的子序列,因此我们需要考虑判重

对于$k<i$,$k$表示最大的$a[k]=a[i]$,显然接了$a[k]$的部分都算重复了

并且重复了$f[k-1]+1$ ($k$前面的以及本身)

因此状态计算还需要变形 :

$f[i]=f[i-1]\ast 2+2-f[pre[a[i]]]-1$

$pre$表示前一个$a[i]$出现的位置,因此我们需要在记录之前先离散化一下

code

const int N = 1e5+10, mod=998244353;
int n,a[N];
int b[N];
int pre[N];
ll dp[N];

void solve(){
	cin>>n;
	for(int i=1;i<=n;i++){
		cin>>a[i];
		b[i] = a[i];
	}
	
	sort(b+1,b+1+n);
	for(int i=1;i<=n;i++){
		a[i] = lower_bound(b+1,b+1+n,a[i])-b;
	}

	
	for(int  i=1;i<=n;i++){
		int t = 0;
		if(pre[a[i]]>0) t = dp[pre[a[i]]-1] +1;
		
		dp[i] = ((dp[i-1]*2+2-t)%mod+mod)%mod;
		
		pre[a[i]] = i ;
		
	}
	
	ll ans =0 ;
	for(int i=1;i<=n;i++){
		ans = (ans+dp[i])%mod;
	}
	cout<<ans<<endl;
	
	
}