Codeforces Round #735 (Div. 2) B. Cobb

该博客主要讨论了一种算法题目,涉及数组操作和位运算。题目要求在给定长度为n的数组中,寻找最大两个元素的OR值,并减去指定参数k与最大AND值的乘积。博主提出了使用二叉树和广度优先搜索(BFS)的方法来求解,通过维护满足特定位运算条件的最大索引,最终计算出答案。博客内容深入探讨了算法实现细节和代码优化技巧。

题目大意

给出一个长度为nnn的数组a1,a2,⋯ ,ana_1,a_2,\cdots,a_na1,a2,,an
i×j−k×(ai∣aj)i\times j - k\times(a_i | a_j)i×jk×(aiaj),其中|表示orkkk是指定参数。

时间限制

1s

数据范围

ai≤n≤105a_i\le n\le 10^5ain105

题解

两个数做or运算,十分不好处理,并不能在二叉树上找出对应的数。
也很难分离出与不与jjj只与iii直接关联的部分。

换一种思路,假设已经确定了ai∣aja_i | a_jaiaj,那么显然就要找到最大两个i,ji,ji,j,而aia_iai并不是很大,于是就可以按照这个思路来处理。

不妨设fsf_sfs表示满足ai&s=aia_i \& s = a_iai&s=ai的最大的iii
gsg_sgs表示满足ai&s=aia_i \& s = a_iai&s=ai的最大的iii
那么转移就可以用一个bfsbfsbfs来实现。
答案就是fs×gs−k×sf_s\times g_s -k\times sfs×gsk×s

Code

//#pragma GCC optimize (2)
//#pragma G++ optimize (2)
#include <cstdio>
#include <algorithm>
#include <cstring>
#include <cmath>
#include <iostream>
#include <queue>
#include <map>
#define ll long long
#define G getchar
using namespace std;

int read()
{
    char ch;
    for(ch = G();ch < '0' || ch > '9';ch = G());
    int n = 0;
    for(;'0' <= ch && ch <= '9';ch = G())n = (n<<1)+(n<<3)+ch-48;
    return n;
}

void write(int x)
{
    if (x > 9)
    {
        write(x / 10);
        putchar(x % 10 + 48);
    }
    else putchar(x + 48);
}

const int N = 131092;

int n , m , k , a[N] , q[N] , f[N] , g[N];
int z[18] , s , K , l , r , S;
bool bz[N];
ll ans , tmp;

int main()
{
    //freopen("b.in","r",stdin);
    //freopen("e.out","w",stdout);
    
    z[0] = 1;
    for (int i = 1 ; i < 18 ; i++)
        z[i] = z[i - 1] << 1;

    for (int T = read() ; T ; T--)
    {
        ans = -1152921504606846976;
        n = read();
        K = read();
        for (k = 0 ; z[k] <= n ; k++);
        m = z[k];
        for (int i = 0 ; i < m ; i++)
        {
            f[i] = g[i] = 0;
            bz[i] = 1;
        }
        for (int i = 1 ; i <= n ; i++)
        {
            a[i] = read();
            if (f[a[i]] == 0) f[a[i]] = i; else
            {
                g[a[i]] = f[a[i]];
                f[a[i]] = i;
            }
        }

        l = 0;
        r = 1;
        q[1] = 0;

        for ( ; l < r ; )
        {
            l++;
            S = q[l];
            if (g[S])
            {
                tmp = f[S];
                tmp = tmp * g[S];
                tmp = tmp - (ll) K * S;
                ans = max(ans , tmp);
            }
            for (int i = 0 ; i < k ; i++)
            {
                if (z[i] & S) continue;
                s = S | z[i];
                if (f[S] > f[s])
                {
                    g[s] = f[s];
                    f[s] = f[S];
                    if (g[S] > g[s]) g[s] = g[S];
                }
                else
                if (f[S] == f[s])
                {
                    if (g[s] < g[S]) g[s] = g[S];
                }
                else
                if (f[S] > g[s]) g[s] = f[S];

                if (bz[s])
                {
                    bz[s] = 0;
                    r++;
                    q[r] = s;
                }
            }
        }

        printf("%lld\n", ans);
    }

    return 0;
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值