Codeforces Round 912 (Div. 2) D2. Maximum And Queries (hard version) 【SOSDP】

吵闹的人群保持笑容多冷静

已于 2024-02-23 08:59:04 修改

阅读量810

点赞数 24

分类专栏： codeforces 练习文章标签：算法 c++ 笔记

于 2023-12-21 10:28:53 首次发布

本文链接：https://blog.csdn.net/m0_73500785/article/details/135123072

版权

codeforces 练习专栏收录该内容

51 篇文章 0 订阅

订阅专栏

D2. Maximum And Queries (hard version)

题意

给定一个长度为 $n$ 的整数数组 $a$ 和一个整数 $k$ ，最多可以执行 $k$ 次操作，每次操作：

选择 $a$ 的一个元素并将其 $+ 1$

求最大的数组按位与，即： $\forall x \in a,ans$ & $= x$

思路

这道题需要用到 $SOS D P$ ， $co d e f orces$ 的这篇 blog 有详细教程。

由于 $a_i \leq10^6$ ，所以当 $\geq \sum (10^6 - a_i)$ 时，根据高位到低位的原则，最终答案一定大于等于 $10^6$ ，因此这时候我们可以充分利用已经有的部分 $\sum a_i$ ，然后将 $k$ 次操作平均分配给每个数，使得每个元素的值尽可能接近，最终答案是： $\lfloor \dfrac{k + \sum a_i}{n} \rfloor$

当 $\sum (10^6 - a_i)$ 时，按照和 $D 1$ 类似的思路贪心从高位到低位拼凑：
对于当前的答案 $x$ 和当前考虑的位 $b$ ，为了将答案 $x$ 变为 $x + 2^b$ ，对于某个 $a_i$ ：

如果 $\not\subseteq a_i$ ，也即是 $x$ 不是 $a_i$ 的子掩码，那么在 $b$ 位的更高位，一定已经进行了若干次 $+$ 的操作，从而将 $a_i 原本为0 的那一位变成1$ ，例如 $x = 1010 , a_i = 1001$ ，此时 $b$ 为最低位，由于 $\not\subset a_i (x_1 = 1 而 a_{i_2} = 0)$ ，因此在之前的操作中，为了将 $a_{i_2}$ 变为 $1$ ，必须经过若干次操作，在这之后， $a$ 的最低位一定会变为 $0$ 。
因此这种情况下要操作的次数是： $2^b$
若 $2^b \subseteq a_i$ ，那么说明 $a_{i_b} = 1$ ，这种情况不需要操作
其他情况下，也即是 $\subseteq a_i \bigwedge x+2^b \not\subseteq a_i$ ，这种情况下 $a_{i_b} = 0$ ，需要的操作数是 $2^b - a_i mod 2^b$ ，也即是充分利用 $a_i$ 的更低位的那些已经有的值，节约操作次数

通过上述的三种情况，不难发现我们需要预处理两种信息：

$c n t [ma s k]$ : 在数组 $a$ 中有多少个值 $a_i$ 满足 $\subseteq a_i$ ，也就是 $ma s k$ 有多少个 父掩码
$d p s u m [ma s k] [b]$ ：对于 $ma s k$ 的所有 父掩码 $x$ ，对他们模 $2^b$ 的值求和

上面两种信息可以用 $SOS$ $D P$ 预处理得到：
与 $b l o g$ 里的不同，这里是对于一个 $ma s k$ 归并父掩码的信息，因此判断某一位 $i$ 的时候， $i f$ 语句要反过来：

if(!(mask & (1<<i)))

如果这一位是 $0$ ，那么它父亲这一位可以是 $0$ 或 $1$ ，因此要加上低 $i - 1$ 位的信息

fore(i,0,20)
    	fore(mask,0,1<<20)
    		if(!(mask & (1<<i)))
    			cnt[mask] += cnt[mask^(1<<i)];

对于 $d p s u m$ 的求解也是类似。

最后对于每次询问，第一种情况需要的操作次数是： $2^b \cdot (n-cnt[ans])$
第三种情况需要的操作次数是： $2^b \cdot(cnt[ans] - cnt[ans|2^b]) - (dpsum[ans][b] - dpsum[ans|2^b][b])$
两种情况加起来就是代码中的表达式了。

完整代码：

// Problem: D2. Maximum And Queries (hard version)
// Contest: Codeforces - Codeforces Round 912 (Div. 2)
// URL: https://codeforces.com/contest/1903/problem/D2
// Memory Limit: 512 MB
// Time Limit: 7000 ms
// 
// Powered by CP Editor (https://cpeditor.org)

#include<bits/stdc++.h>
#define fore(i,l,r)	for(int i=(int)(l);i<(int)(r);++i)
#define fi first
#define se second
#define endl '\n' 
#define ull unsigned long long

const int INF=0x3f3f3f3f;
const long long INFLL=0x3f3f3f3f3f3f3f3fLL;

typedef long long ll;

const int N = 1000050;

ll cnt[1<<20]; //how many elements from the array is mask a submask of
ll dpsum[1<<20][20]; //sum of a_i mod 2^b over all ai for which x is a submask

int main(){
    std::ios::sync_with_stdio(false);
    std::cin.tie(nullptr);
    std::cout.tie(nullptr);
    int n,q;
    std::cin>>n>>q;
    std::vector<int> a(n+1);
    ll div = 0 , sum = 0;
    fore(i,1,n+1){
    	std::cin>>a[i];
    	sum += a[i];
    	div += ((1ll<<20) - a[i]);
    	++cnt[a[i]]; //本身就是自己的父掩码
    	ll val = 0;
    	fore(j,0,20){
    		dpsum[a[i]][j] += val; //本身就是自己的父掩码
    		val += (a[i] & (1ll<<j));
    	}
    }
    
    fore(i,0,20)
    	fore(mask,0,1<<20)
    		if(!(mask & (1<<i)))
    			cnt[mask] += cnt[mask|(1<<i)];
    		
    fore(i,0,20)
    	fore(mask,0,1<<20)
    		if(!(mask & (1<<i)))
    			fore(p,0,20)
    				dpsum[mask][p] += dpsum[mask|(1<<i)][p];

    while(q--){
    	ll k;
    	std::cin>>k;
    	if(k >= div){
    		std::cout<<(k+sum)/n<<endl;
    		continue;
    	}
    	ll ans = 0;
    	for(int i=19;i>=0;--i){
    		ll opt = (n - cnt[ans|(1<<i)]) * (1ll<<i) - (dpsum[ans][i] - dpsum[ans|(1<<i)][i]);
    		if(opt <= k){
    			k -= opt;
    			ans |= 1<<i;
    		}
    	}
    	std::cout<<ans<<endl;
    }
	return 0; 
}