[ AT_agc009_c]Division into Two

题目传送门

大家都知道集合吧,原谅我喜欢说废话

解法

先钦定 A > B A>B A>B,

先把数排好序得到数组 a a a,考虑先解决集合 X X X的问题,

设计状态:

明显只有一维( n < = 1 e 5 n<=1e5 n<=1e5),所以有:
f i : X 集合最后放的是第 i 个数,对于前 i 个数来说 X , Y 都合法的方案数 f_i:X集合最后放的是第i个数,对于前i个数来说X,Y都合法的方案数 fi:X集合最后放的是第i个数,对于前i个数来说X,Y都合法的方案数

发现后 ( n − i ) (n-i) (n−i)个放在 Y Y Y的数我们未保证合法,但是我们之后特判一下就好了

状态转移

设 f i f_i fi由 f j f_j fj转移而来

考虑有哪些 j j j可以贡献:

  1. i > j i>j i>j(废话)

  2. a i − a j ≥ A a_i-a_j\ge A ai−aj≥A(差值大于 A A A)

  3. max ⁡ { a k − a k − 1 } ≥ B , k ∈ ( j , i ) \max\{a_k-a_{k-1}\}\ge B,k\in(j,i) max{ak−ak−1}≥B,k∈(j,i)(满足 Y Y Y集合的要求)

则: f i = ∑ f j f_i=\sum f_j fi=∑fj

code:

复制代码
#include <iostream>
#include <algorithm>
using namespace std;
typedef long long ll;
const int mod = 1e9 + 7,N=1e5+7;
int n,l,r;
int f[N],sum[N];
ll A,B;
ll a[N];
int main(){
	scanf("%d%lld%lld",&n,&A,&B);
	if(A<B) swap(A,B);
	for(int i=1;i<=n;i++) scanf("%lld",&a[i]);
	for (int i=1;i+2<=n;++i)
        if (a[i+2]-a[i]<B) return puts("0"),0;
	a[0]=0;
	f[0]=sum[0]=1;
	for(int i=1;i<=n;i++) {
		while(a[i]-a[r+1]>=A) r++;
		if(l<=r) {
			f[i]=(1ll*f[i]+1ll*sum[r])%mod;
			if(l) f[i]=(1ll*f[i]-1ll*sum[l-1]+mod)%mod;
		}
		sum[i]=(1ll*sum[i-1]+1ll*f[i])%mod;
		if(a[i]-a[i-1]<B) l=i-1;
	}
	int ans=0;
	for(int i=n;~i;i--){	
		ans=(1ll *ans+1ll*f[i])	%mod;
		if(i<n && a[i+1]-a[i] <B) break;
	}
	printf("%d\n",ans);
}

TXL

相关推荐
爱睡懒觉的焦糖玛奇朵7 小时前
【从视频到数据集:焦糖玛奇朵的魔法工具使用说明】
人工智能·python·深度学习·学习·算法·yolo·音视频
Runawayliquor7 小时前
opbase:CANN 所有算子的公共地基
大数据·数据库·人工智能·算法
徐安安ye7 小时前
FlashAttention 为什么对序列长度这么“敏感”?
人工智能·算法
潜创微科技7 小时前
IT6520:USB‑C 转 MIPI 芯片方案 4K@120Hz 高清显示
c语言·开发语言
黎阳之光8 小时前
黎阳之光:以视频孪生重构智能监盘,为燃机打造新一代智慧电厂大脑
大数据·人工智能·算法·安全·数字孪生
绝知此事9 小时前
【算法突围 02】树形结构与数据库索引:树形结构与数据库索引:从 BST 到 B+ 树的演化与 MySQL 优化
数据库·mysql·算法·面试·b+树
清木!9 小时前
排序算法比较
数据结构·算法·排序算法
hh.h.10 小时前
CANN算子开发入门:从零开始写第一个Ascend C算子
c语言·开发语言·cann·c算子
吴可可12310 小时前
用Teigha修改并保存CAD文件
数据库·算法·c#
AI科技星10 小时前
全域数学·第三部·数术几何部·平行网格卷 完整专著目录(含拓扑发展史+学科定位·终稿)
c语言·开发语言·网络·量子计算·agi