牛客第七场 H Pair —— 数位dp

最新推荐文章于 2021-01-28 20:28:43 发布

Scar_Halo

最新推荐文章于 2021-01-28 20:28:43 发布

阅读量236

点赞数 1

分类专栏： ACM - 数位DP 文章标签：牛客第七场 H Pair 数位dp

本文链接：https://blog.csdn.net/Scar_Halo/article/details/101110986

版权

ACM - 数位DP 专栏收录该内容

3 篇文章 0 订阅

订阅专栏

题目链接：点我啊╭(╯^╰)╮

题目大意：

每次加入一个区间的数，求中位数

解题思路：

    考虑到高位决定结果，所以从高位开始数位dp
     $o a$ 与 $o b$ 表示 $a$ 和 $b$ 是否到达了上界
     $o a n d$ 与 $o x o r$ 表示高位的和与异或是否出现非零
     $z a$ 与 $z b$ 限制正数

核心：两个数同时数位dp

#include<bits/stdc++.h>
#define rint register int
#define deb(x) cerr<<#x<<" = "<<(x)<<'\n';
using namespace std;
typedef long long ll;
typedef pair <int,int> pii;
const ll mod = 998244353;
const int maxn = 2e6 + 10;
int T, A, B, C;
int a[40], b[40], c[40];
ll dp[40][2][2][3][3][2][2];

ll dfs(int wz, int oa, int ob, int oand, int oxor, int za, int zb){
	ll& ret = dp[wz][oa][ob][oand+1][oxor+1][za][zb];
	if(ret != -1) return ret;
	if(wz == 0){
		if((oand>0 || oxor<0) && za && zb) return ret = 1;
		return ret = 0;
	}
	wz--, ret = 0;
	for(int i=0; i<=(oa?1:a[wz]); i++)
		for(int j=0; j<=(ob?1:b[wz]); j++)
			ret += dfs(wz, oa|(i^a[wz]), ob|(j^b[wz]), oand?oand:(i&j)-c[wz],\
					 oxor?oxor:(i^j)-c[wz], za|i, zb|j);
	return ret;
}

int main() {
	scanf("%d", &T);
	while(T--){
		scanf("%d%d%d", &A, &B, &C);
		memset(dp, -1, sizeof(dp));
		for(int i=30; ~i; i--){
			a[i] = (A >> i) & 1;
			b[i] = (B >> i) & 1;
			c[i] = (C >> i) & 1;
		}
		printf("%lld\n", dfs(31,0,0,0,0,0,0));
	}
}