作业【状压DP】

最新推荐文章于 2024-10-27 18:37:08 发布

ssl_fuyang

最新推荐文章于 2024-10-27 18:37:08 发布

阅读量77

点赞数

分类专栏： DP 文章标签： c语言动态规划

本文链接：https://blog.csdn.net/qq_43010386/article/details/120512299

版权

DP 专栏收录该内容

125 篇文章 1 订阅

订阅专栏

这篇博客介绍了一个关于作业安排的算法问题，数据范围较小，适合采用动态规划的思路。博主首先分析了题目，指出可以使用状态压缩动态规划（状压DP）来解决。在DP过程中，博主预处理了0的前缀和，以便在O(1)时间内查询某个状态下某段区间0的个数。最后，博主给出了完整的C++代码实现，并计算出了答案。

摘要由CSDN通过智能技术生成

>Link

luogu U137979

>Description
在这里插入图片描述
对于 100% 的数据， $1 ≤b_i ≤ n ≤ 18, 1 ≤ a_i ≤ 10$

>解题思路

一开始没看数据范围想了好久QAQ

然后一看这个数据范围应该就是状压DP
设 $f_{s,i}$ 为在 $s$ 状态下，最后一个写的作业为第 $i$ 个的方案数，状态 $s$ 记录对应的作业有没有写
直接按照题意暴力判断当前枚举到的 $s$ 和 $i$ 是否合法
第二个条件我们发现是求当前 $s$ 状态下，某段区间的0的个数，可以预处理每个状态下0的前缀和，DP时 $O (1)$ 查询

>代码

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#define N 25
using namespace std;

const int Mod = 4921057;
int n, a[N], b[N], sum[N], f[300000][N], T, minn, ans;
int tot[300000][N];

int main()
{
	int cnt, l, r; bool ok;
	minn = 10;
	scanf ("%d", &n);
	for (int i = 1; i <= n; i++)
	  scanf ("%d", &a[i]), sum[a[i]]++, minn = min (minn, a[i]);
	for (int i = 1; i <= n; i++) scanf ("%d", &b[i]);
	for (int i = 1; i <= 10; i++) sum[i] += sum[i - 1];
	
	T = (1 << n) - 1;
	for (int i = 1; i <= T; i++)
	  for (int j = 1; j <= n; j++)
	  {
	  	tot[i][j] = tot[i][j - 1];
	  	if (((i >> (j - 1)) & 1) == 0) tot[i][j]++;
	  }
	for (int i = 1; i <= n; i++)
	  if (a[i] == minn) f[1 << (i - 1)][i] = 1;
	for (int i = 1; i <= T; i++)
	  for (int j = 1; j <= n; j++)
	  {
	  	if (((i >> (j - 1)) & 1) == 0) continue;
	  	cnt = 0, ok = 1;
	  	for (int jj = 1; jj <= n; jj++)
	  	  if ((i >> (jj - 1)) & 1)
	  	  {
	  	  	if (a[jj] > a[j]) {ok = 0; break;}
	  	  	cnt++;
		  }
		if (!ok || cnt <= sum[a[j] - 1]) continue;
		int ii = i ^ (1 << (j - 1));
		for (int jj = 1; jj <= n; jj++)
		{
	  		if (((ii >> (jj - 1)) & 1) == 0) continue;
	  		/*cnt = 0;
	  		for (int t = min (j, jj) + 1; t < max (j, jj); t++)
	  		  if (((ii >> (t - 1)) & 1) == 0) cnt++;
	  		if (cnt <= b[jj]) f[i][j] = (f[i][j] + f[ii][jj]) % Mod;*/
	  		l = min (j, jj) + 1, r = max (j, jj) - 1;
	  		if (tot[ii][r] - tot[ii][l - 1] <= b[jj])
	  		  f[i][j] = (f[i][j] + f[ii][jj]) % Mod;
		}
	  }
	for (int i = 1; i <= n; i++)
	  ans = (ans + f[T][i]) % Mod;
	printf ("%d", ans);
	return 0;
}