首页 > 其他分享 >[NOI Online 2020 #2 提高 B] 子序列问题 (dp+线段树)

[NOI Online 2020 #2 提高 B] 子序列问题 (dp+线段树)

时间:2022-10-28 18:33:40浏览次数:67  
标签:last NOI 线段 times leq 2020 Online +...+ dp

真的是事后诸葛亮,一下考场就知道怎么做了,现在把题解补回来

看到这个问题,我第一下想到的就是 \(dp\)。

如何 \(dp\)?

设 \(dp[i]\) 为以 \(i\) 为右端点的所有子串的答案之和,那么最后要输出的答案就是 \(ans=\sum_{i=1}^{n}dp[i]\)。

至于怎么维护这个 \(dp[i]\),我一开始想的做法也很简单,直接暴力。

也就是说,对于每一个右端点 \(r\),从右向左枚举左端点 \(l\),然后统计答案。

但是如何维护 \(f(l,r)\) 呢?

经典套路:对于每一个数,我们把它出现在序列 \([1,n]\) 中最右边的那个位置找出来。对于数 \(x\),记为 \(last[x]\)。

假设我们现在已经求出了 \(f(i,r)\),然后要求 \(f(i-1,r)\),那么显然,若 \(last[a[i-1]]\not=i\),就说明还有一个与 \(a[i-1]\) 相同的数出现在位置 \(i-1\) 的右边,则 \(f(i-1,r)=f(i,r)\)。否则,若 \(last[a[i-1]]=i\),则 \(f(i-1,r)=f(i,r)+1\)。

然后求完 \(dp[i]\) 后记得更新 \(last\)。

那么这种方法是 \(O(n^2)\) 的,只能拿 \(50\) 分。

如何提高?

我们回想一下刚才加粗的这一段:

假设我们现在已经求出了 \(f(i,r)\),然后要求 \(f(i-1,r)\),那么显然,若 \(last[a[i-1]]\not=i\),就说明还有一个与 \(a[i-1]\) 相同的数出现在位置 \(i-1\) 的右边,则 \(f(i-1,r)=f(i,r)\)。否则,若 \(last[a[i-1]]=i\),则 \(f(i-1,r)=f(i,r)+1\)。

考虑如何优化?

先画个图:

在这里插入图片描述
整个图的下方的方格表示的是序列,方格上面箭头指着的数是这个位置的编号,方格内的数代表这个位置的数。

如图,其中 \(x\) 代表数 \(a[i]\),然后我们找到 \(last[x]\) 的位置,那么显然,在序列 \([last[x]+1,i-1]\) 中,每一个数都与 \(x\) 不同,不妨记为 \(o\)(注意:每一个 \(o\) 之间有可能不同);在序列 \([1,last[x]-1]\) 中,每一个数都有可能与 \(x\) 相同,也可能不同,不妨记为 \(a\)。

然后我们现在要求出每一个 \(f(l,i)\),\((1\leq l\leq i)\)。

分情况讨论:

  1. \(1\leq l\leq last[x]\)

    在这里插入图片描述
    我们就以这幅图为例。我们对比一下两段序列(红和蓝),发现红色的只是比蓝色的多了一个 \(x\),而这个多出来的 \(x\) 并没有贡献。

    所以在这种情况下,必有 \(f(l,i)=f(l,i-1)\)。

    平方后也相等。

  2. \(last[x]<l<i\)

    在这里插入图片描述
    还是看图。

    你会发现,红框还是比蓝框多了个 \(x\),但是由于 \(o\) 是不可能等于 \(x\) 的,所以这个 \(x\) 有贡献。

    那么 \(f(l,i)=f(l,i-1)+1\)

    平方后 \(f(l,i)^2=f(l,i-1)^2+2\times f(l,i-1)+1\)

  3. \(l=i\),\(f(l,i)=1\)

把三个讨论整合一下:

\[\begin{aligned} dp[i]=&f(1,i)^2+f(2,i)^2+...+f(last[x],i)^2+f(last[x]+1,i)^2+...+f(i-1,i)^2+f(i,i)^2\\ =&f(1,i-1)^2+f(2,i-1)^2+...+f(last[x],i-1)^2+[f(last[x]+1,i-1)^2\\ &+2\times f(last[x]+1,i-1)+1]+...+[f(i-1,i-1)^2+2\times f(i-1,i-1)+1]+1\\ =&[f(1,i-1)+f(2,i-1)+...+f(i-1,i-1)]\\ &+2\times[f(last[x]+1,i-1)+f(last[x]+2,i-1)+...+f(i-1,i-1)]+(i-last[x]-1)+1\\ =&dp[i-1]+2\times[f(last[x]+1,i-1)+f(last[x]+2,i-1)+...+f(i-1,i-1)]+(i-last[x])\\ =&dp[i-1]+2\times \sum_{l=last[x]+1}^{i-1}f(l,i-1)+(i-last[x]) \end{aligned}\]

那么,我们就可以通过维护 \(\sum_{l=last[x]+1}^{i-1}f(l,i-1)\) 来求出 \(dp[i]\)。

如何维护?

考试的时候就是想到这就想不下去了

这个时候,我们就会用到一颗线段树。

这棵线段树用来维护每一个 \(f(l,i-1)\) 的值的和,其中 \(1\leq l<i\)。

那么我们可以通过询问线段树中的区间 \([last[x]+1,i-1]\) 就可以知道 \(\sum_{l=last[x]+1}^{i-1}f(l,i-1)\) 是什么了。

然后考虑求完 \(dp[i]\) 如何更新线段树。

由于我们刚才已经得到:

  1. 当 \(1\leq l\leq last[x]\) 时,有 \(f(l,i)=f(l,i-1)\),所以这一段不用管它,还是原来的值。

  2. 当 \(last[x]<l<i\) 时,有 \(f(l,i)=f(l,i-1)+1\),所以我们只用把这一段的值整体加上 \(1\) 就好了。

  3. 当 \(l=i\) 时,有 \(f(l,i)=1\),所以我们也把这个点修改成 \(1\),这一个步骤可以和前一个并在一起。

这样就能更新所有的 \(f()\) 了。

总时间复杂度 \(O(n\log n)\),需要注意一下细节(比如离散化和卡常)。

代码如下:

#include<bits/stdc++.h>

#define N 1000010
#define mod 1000000007
#define ll long long

using namespace std;

int n,a[N],b[N];
int last[N];
ll dp[N],sum[N<<2],lazy[N<<2];

void down(const int k,const int l,const int r,const int mid)
{
	if(lazy[k])
	{
		sum[k<<1]+=lazy[k]*(mid-l+1);
		sum[k<<1|1]+=lazy[k]*(r-mid);
		lazy[k<<1]+=lazy[k];
		lazy[k<<1|1]+=lazy[k];
		lazy[k]=0;
	}
}

ll query(const int k,const int l,const int r,const int ql,const int qr)
{
	if(ql>qr) return 0;
	if(ql<=l&&r<=qr) return sum[k];
	const int mid=(l+r)>>1;
	ll ans=0;
	down(k,l,r,mid);
	if(ql<=mid) ans+=query(k<<1,l,mid,ql,qr);
	if(qr>mid) ans+=query(k<<1|1,mid+1,r,ql,qr);
	return ans;
}

void update(const int k,const int l,const int r,const int ql,const int qr)
{
	if(ql<=l&&r<=qr)
	{
		sum[k]=sum[k]+r-l+1;
		lazy[k]++;
		return;
	}
	const int mid=(l+r)>>1;
	down(k,l,r,mid);
	if(ql<=mid) update(k<<1,l,mid,ql,qr);
	if(qr>mid) update(k<<1|1,mid+1,r,ql,qr);
	sum[k]=sum[k<<1]+sum[k<<1|1];
}

int read()
{
	int x=0;
	char ch=getchar();
	while(ch<'0'||ch>'9') ch=getchar();
	while(ch>='0'&&ch<='9')
	{
		x=(x<<1)+(x<<3)+(ch^'0');
		ch=getchar();
	}
	return x;
}

int main()
{
	n=read();
	for(register int i=1;i<=n;i++)
		b[i]=a[i]=read();
	sort(b+1,b+n+1);
	const int nn=unique(b+1,b+n+1)-b-1;
	for(register int i=1;i<=n;i++)
		a[i]=lower_bound(b+1,b+nn+1,a[i])-b;//离散化
	for(register int i=1;i<=n;i++)
	{
		const ll len=query(1,1,n,last[a[i]]+1,i-1);
		dp[i]=(0ll+dp[i-1]+2*len+(i-last[a[i]]-1)+1)%mod;//计算dp[i]
		update(1,1,n,last[a[i]]+1,i);//更新f
		last[a[i]]=i;//记得更新last
	}
	ll ans=0;
	for(register int i=1;i<=n;i++)
		ans=(ans+dp[i])%mod;//求和
	printf("%lld\n",ans);
	return 0;
}

至于如何卡常建议去看一下洛谷讨论区。

标签:last,NOI,线段,times,leq,2020,Online,+...+,dp
From: https://www.cnblogs.com/ez-lcw/p/16837043.html

相关文章

  • [NOI Online #3 提高组] 魔法值(矩阵快速幂+各种优化)
    矩阵快速幂相信其他巨佬讲矩阵快速幂的原理和过程都已经很详细了,这里我就着重讲一讲用裸的矩阵快速幂算法之后,如何优化。优化1:优化矩阵乘法这个好几位大佬讲过了,我也不......
  • NOIP 前 CF*2300 左右 dp 做题记录
    iFTL独立做出形象理解为两个横放的羽毛球盒,第一个放长t1的,第二个放长t2的,然后手推可得一次共同发射之后必是平平的盒子底部,仿佛回到了最初,至此可发现子结构。基于此......
  • ON1 NoNoise AI 2023 for mac(图片降噪软件)中文版
    ON1NoNoiseAIformac一款图片降噪软件,ON1NoNoiseAI中文版智能去除所有图像噪点,同时智能恢复和增强细节。它通过支持常见的照片编辑器和文件格式集成到您的工作流程中,......
  • 洛谷 P1077 [NOIP2012 普及组] 摆花 (DP)
    https://www.luogu.com.cn/problem/P1077题目描述摆上m盆花。一共有n种花,从1到n标号。为了在门口展出更多种花,规定第i种花不能超过ai盆,摆花时同一种花放在一起,且不同......
  • [Ynoi2016] 掉进兔子洞
    \(\text{Solution}\)莫队配合\(\text{bitset}\)发现答案困难的部分在于同一个数在三个区间出现次数的最小值考虑强行拆开看,用莫队处理出每个区间每个数的出现次数,这个......
  • [线段树套单调栈] 2019-2020 ICPC Asia Hong Kong Regional Contest H.[Hold the Line
    线段树套单调栈2019-2020ICPCAsiaHongKongRegionalContestH.​​HoldtheLine​​题目大意你已经建造了一条由编号从到的战壕组成的防线,每条战壕最初都是空的。士......
  • 2022NOIPA层联测16
    数塔:相等上传非常显然,重点是怎么二分(对于这种不知道更大的更优还是更小的更优的题,不知道选哪个二分模板。。)大于等于和小于等于都可以,重要的是取等,就是保证答案在二分......
  • 洛谷P3225 [HNOI2012]矿场搭建
    SLOJH7136.「HNOI2012」矿场搭建题目描述煤矿工地可以看成是由隧道连接挖煤点组成的无向图。为安全起见,希望在工地发生事故时所有挖煤点的工人都能有一条出路逃到救援......
  • 20201318李兴昕第六章学习笔记
    第五章:定时器及时钟服务知识点归纳总结:本章讲述了信号和信号处理;介绍了信号和中断的统一处理,有助于从正确的角度看待信号;将信号视为进程中断,将进程从正常执行转移到......
  • noi.cn 访问量爬取
    网课期间开始的一项无聊的项目,对noi.cn的访问量进行爬取。具体操作为直接访问对应的网址,获取其网站底部的总访问量信息。爬虫使用Python编写,配合bat文件和Windows......