模拟赛20210809A T1

模拟赛20210809A T1

题目

大意：一个长度为$n$的序列，求经过$n-1$次合并操作后所有可能的结果（从小到大输出）
定义合并操作为：每次操作可将任意一对相邻的数$a,b$ 合并成$((a\&b)+(a|b)>>1$。

思路

区间DP：
设 $f_{i,j,k}$ 表示 $i$ 到 $j$ 这个区间内能否经过若干次合并得到 $k$，能为1，不能为0。
因为$((a\&b)+(a|b)>>1$ 实际上等于$(a+b)/2$，而 $a_i$最大为7，所以所有结果都不会超过7。

第一层循环枚举区间长度
第二层枚举区间位置（一个端点）
第三层枚举中间点$mid$，表示将区间分为左区间$[i,mid]$ 和右区间$[mid+1,j]$。
第四层枚举左区间能合并成的结果 $p$，即$f[i][mid][p]==1$
第五层枚举右区间能合并成的结果 $q$，即$f[mid+1][j][q]==1$

因为左区间能合并成 $p$，右区间能合并成 $q$，所以整个区间能合并成$(p+q)/2$
即当 $f[i][mid][p]==1$ 且 $f[mid+1][j][q]==1$ 时 ，$f[i][j][(p+q)/2]=1$
最后判断$f_{1,n,i}$中可以合并的数。

因为$1<=i<=mid<j<=n$，$0<=k<=7$
所以时间复杂度为跑不满的$O(n^3{8}^2)=O(64n^3)，对于此题n<=150是可以过的$

代码

#include<iostream>
#include<cstdio>
#define reg register
using namespace std;
int n,a[160];
bool f[160][160][10];
int mer(int a,int b)//合并操作
{
	return (((a&b)+(a|b))>>1);
}
int main()
{
	scanf("%d",&n);
	for(reg int i=1;i<=n;++i)
	{
		scanf("%d",&a[i]);
		f[i][i][a[i]]=true;//初始化
	}
	for(reg int i=2;i<=n;++i)//枚举区间长度
	{
		for(reg int j=i;j>=1;--j)//枚举区间左端点（要倒着）
		{
			for(reg int mid=j;mid<i;++mid)//枚举中间点
			{
				for(reg int l=0;l<=7;++l)//枚举左区间能不能合并的数
				{
					if(!f[j][mid][l]) continue;//不能就跳过
					for(reg int r=0;r<=7;++r)//枚举右区间能不能合并的数
					{
						if(!f[mid+1][i][r]) continue;//不能就跳过
						int t=mer(l,r);//合并操作
						f[j][i][t]=true;
					}
				}
			}
		}
	}
	for(reg int i=0;i<=7;++i)
	{
		if(f[1][n][i]) printf("%d ",i);//判断并输出
	}
}