CCF-CSP 202104-4 校门外的树(DP/好题)

问题描述

X 校最近打算美化一下校园环境。前段时间因为修地铁,X 校大门外种的行道树全部都被移走了。现在 X 校打算重新再种一些树,为校园增添一抹绿意。

X 校大门外的道路是东西走向的,我们可以将其看成一条数轴。在这条数轴上有 n 个障碍物,例如电线杆之类的。虽然障碍物会影响树的生长,但是障碍物不一定能被随便移走,所以 X 校规定在障碍物的位置上不能种树。n 个障碍物的坐标都是整数;如果规定向东为正方向,则 n 个障碍物的坐标按照从西到东的顺序分别为 a1,a2,⋯,an。X 校打算在 [a1,an] 之间种一些树,使得这些树看起来比较美观。

X 校希望,在一定范围内,树应该是等间隔的。更具体地说,如果把 [a1,an) 划分成一些区间 [ap1,ap2),⋯,[apm−1,apm)(1=p1<p2<⋯<pm=n),那么每个区间 [api,api+1) 内需要至少种一棵树,且该区间内种的树的坐标连同区间端点 api,api+1 应该构成一个等差数列。不同区间的公差,也就是树的间隔可以不相同。

例如,如果障碍物位于 0,2,6 这三处,那么我们可以选择在 [0,2) 和 [2,6) 分别种树,也可以选择在 [0,6) 等间隔种树。如果是分别在 [0,2) 和 [2,6) 种树,由于每个区间内至少要种一棵树,坐标 1 上必须种树;而 [2,6) 上的树可以按照 1 的间隔种下,也可以按照 2 的间隔种下。下图表示了这两种美观的种树方案,其中橙色的圆表示障碍物,绿色的圆表示需要在这个位置种树,箭头上的数字表示种下这棵树时对应的间隔为多少。

sample1_h.png
对区间 [0,2) 和 [2,6) 分别以 1 和 2 的间隔种树是美观的

sample2_h.png
对区间 [0,2) 和 [2,6) 分别以 1 的间隔种树也是美观的

而如果选择在 [0,6) 区间等间隔种树,我们只能以 3 的间隔种树,因为无论是选择间隔 1 或者间隔 2,都需要在坐标 2 上种树,而这个位置已经有障碍物了。下图分别表示了间隔为 3,2,1 时的种树情况,红色箭头表示不能在这里种树。

sample3_h.png
对区间 [0,6) 以 3 的间隔种树是美观的

sample4_h.png
对区间 [0,6) 以 2 的间隔种树是不美观的

sample5_h.png
对区间 [0,6) 以 1 的间隔种树也是不美观的

一般地,给定一个区间 [al,ar),对于树的坐标的集合 T⊂(al,ar)(T⊂Z),归纳定义 T 在 [al,ar) 上是美观的

  1. 如果 T≠∅,T∩{al,al+1,⋯,ar}=∅,并且存在一个公差 d≥1,使得 T∪{al,ar} 中的元素按照从小到大的顺序排序后,可以构成一个公差为 d 的等差数列(显然,这个等差数列的首项为 al,末项为 ar),则 T 在 [al,ar) 上是美观的;
  2. 如果 T∩{al,al+1,⋯,ar}=∅,并且存在一个下标 m(l<m<r),使得 T∩(al,am) 在 [al,am) 上是美观的,且 T∩(am,ar) 在 [am,ar) 上是美观的,则 T 在 [al,ar) 上是美观的。

根据这一定义,空集在任意区间上都不是美观的;另外,如果存在下标 i 使得 ai∈T,那么 T 一定不是美观的。

我们称两种种树的方案是本质不同的,当且仅当两种方案中,种树的坐标集合不同。请帮助 X 校对 [a1,an) 求出所有本质不同的美观的种树方案。当然,由于方案可能很多,你只需要输出总方案数对 109+7 取模的结果。

输入格式

输入的第一行包含一个正整数 n,表示障碍物的数量。

输入的第二行包括 n 个非负整数 a1,⋯,an,表示每个障碍物的坐标。

保证对 i=1,2,⋯,n−1,ai<ai+1。

输出格式

输出一个非负整数,表示本质不同的美观的种树方案的数量对 109+7 取模的结果。

样例输入

3
0 2 6

Data

样例输出

3

Data

样例说明

这组样例即为题面描述中提到的那组。

样例输入

11
0 10 20 30 40 50 60 70 80 90 100

Data

样例输出

256507

Data

样例输入

333
33 44 67 210 528 762 873 984 1234 1466 1739 2859 3421 4061 4598 5172 5201 5220 5261 5322 5389 5559 6670 7070 7898 8079 8129 8192 8616 8641 8806 9559 9585 9750 10263 10627 10674 10692 10903 11649 11885 12179 12307 12743 13173 13352 13389 13496 13611 15292 15321 16018 16327 16415 16959 16972 17499 17617 17786 18476 18966 19239 19498 19875 20312 20392 21603 21620 21730 21967 21972 21999 22015 22590 22775 23709 23839 24165 24408 24595 25160 25479 25812 26482 27328 28101 28297 28305 28342 28557 28986 29110 29401 29765 30292 30493 30739 31027 31201 31218 31414 32089 32759 32770 32777 32815 32877 32890 33297 33457 33603 33757 33866 34498 34525 34659 34679 34861 34870 34997 35311 35846 36411 36457 36738 36902 37940 38228 40156 40320 40705 40737 40803 41066 41443 41460 41954 41968 42040 42062 42099 43281 43320 43527 43537 43587 43729 44750 44822 45655 45769 46109 46525 47060 47128 47999 48635 48887 48981 49366 49424 49524 50546 50580 50689 51332 51861 51943 52097 52702 53009 53067 53397 53526 53901 54280 54399 54801 55535 55592 55740 55843 56110 56428 56552 56682 56848 57179 57688 57797 57847 57959 58330 58831 59553 59699 59884 59939 61233 61636 61732 61908 62145 62549 62649 62740 62912 62971 63053 64312 64322 64412 64816 64845 64873 64923 64976 65023 65166 65496 66065 66491 66803 66941 67081 68331 68336 68360 68476 69179 69719 69758 69948 70072 70544 70598 70990 71014 71454 71687 71743 71958 72282 72384 72456 72985 73327 74325 75046 75097 76647 77062 77088 77431 77553 77673 77753 78217 78518 78564 79565 79588 79686 80275 80939 81052 81348 81386 81440 81589 81610 81793 82408 82801 82836 83239 83466 83610 83867 83943 84441 84467 85248 85305 85554 85565 85758 86251 86603 86743 87323 87565 87824 87833 88265 88309 89178 89509 89618 89699 89708 90331 90359 90878 90902 91449 92284 92374 92549 92609 93609 94345 94934 95140 95475 95733 95985 95995 96270 96641 96807 97003 97632 98160 98677 98853 98943 99037 99055 99075 99185 99395 99592

Data

样例输出

7094396

Data

评测用例规模与约定

对于 10% 的数据,保证 n=2;

对于 30% 的数据,保证 n≤10;

对于 60% 的数据,保证 n≤100,ai≤1000;

对于 100% 的数据,保证 2≤n≤1000,0≤ai≤100,000,且至少存在一种美观的种树方案。

不难打出 (n^2mlogm)的暴力dp然后喜提60分。设dp[i]为到第i个障碍物的方案总数,有转移方程(dp[i] = Sigma_{j = 1}^{i - 1}dp[j] imes calc(j, i)),其中calc(j, i)为第j个障碍物到第i个障碍物之间的方案数。关键在于怎么优化calc函数。 首先第j个障碍物到第i个障碍物之间的间隔必须为a[i] - a[j]的因子(显然),方案数一定小于等于因子个数,其次这个间隔还不能撞上障碍物。不妨换个角度,倒着从i - 1开始枚举j,这样一开始i - 1和i之间是没有障碍物的,则a[i] - a[i - 1]的所有因子都满足条件,更新答案并把它们添加到一个集合中;然后到了i - 2,同样先枚举a[i] - a[i - 2]的所有因子,这些因子中已经处于集合中的一定不可,因为按这样的间隔排列一定会有树遇上a[i - 1]这个障碍物。因此倒着枚举到a[i] - a[k]因子的时候如果已经在集合则跳过,如果不在集合则更新数量同时把这个因子插入集合。不要忘记外循环每一次都要先把集合清空,以及要筛法预处理出1e5范围内每个数的因子。

#include <bits/stdc++.h>
#define int long long
#define mod 1000000007
using namespace std;
int n, a[1005], dp[1005];//a数组是单调递增的
set<int> st;
vector<int> f[100005];
int calc(int p, int q) {//从a[x]到a[y]的可能的方案数(中间可能有障碍
	int x = a[q] - a[p];
	int ans = 0;
	//!
	st.insert(x);
	for(int i = 0; i < f[x].size(); i++) {
		if(st.find(f[x][i]) != st.end()) {
			continue;
		} else {
			ans++;
			st.insert(f[x][i]);
		}
	}
	return ans;
}
signed main() {
	ios::sync_with_stdio(false);
	cin >> n;
	for(int i = 1; i <= n; i++) {
		cin >> a[i];
	}
	for(int i = 1; i <= a[n] / 2; i++) {
		for(int j = 2 * i; j <= a[n]; j += i) {
			f[j].push_back(i);
		}
	}
	dp[1] = 1;
	for(int i = 1; i <= n; i++) {
		st.clear();
		for(int j = i - 1; j >= 1; j--) {
			dp[i] = (dp[i] + dp[j] * calc(j, i) % mod) % mod;
		}
	}
	cout << dp[n] % mod;
	return 0;
}

原文地址:https://www.cnblogs.com/lipoicyclic/p/15020078.html