CCPC-Wannafly Winter Camp Day7 (Div2, onsite) F 逆序对! (01字典树的O(nlogn)做法)

思路:

这道题其实还是偏思维,用O(n*n*logn)和O(n*n)的做法都可以过,但核心思想是一样的。

对于任意两个数x,y(x!=y),不妨设x<y 要想使另一个数z使得x^z>y^z的条件是

设 t为 x^y 的最高为1的位(则x的第t位为0,y的第t位为1)

则z的第t位为1。

好好思考一下上述过程。

所以要求1~m中有多少个数能改变x y的大小关系,只需要求有多少个数的第(x^y的最高位)位为1就可以了。

至于怎么求1~m种第k位为1个数字个数,从最高位开始算就行了,具体细节留给读者考虑。

因此,我们只需要求出初始逆序对数,再乘m,再加上每一位为1时对整个数组逆序对变化的贡献就行了。

在zqc大佬的指点下,我才想起01字典树可以O(nlogn)做这个题。于是赶紧敲了一发,中途有一细节错误debug了好久,应该是字典树好久没写了有些不熟悉了吧。。。

代码:

#include<bits/stdc++.h>
#define ll long long
#define inf 0x3f3f3f3f
#define p pair<int,int>
using namespace std;
const int maxn=2010;
const ll mo=998244353;
ll n,m,k;
ll a;
ll sum[65],tmp,ans;
struct Trie
{
    ll ch[maxn*maxn][2],sz;
    ll val[maxn*maxn];// 注意是maxn*maxn RE了三次我才发现
    ll ad[maxn];
    void init()
    {
        memset(ch,0,sizeof(ch));
        sz=0;
        memset(val,0,sizeof(val));
        memset(ad,0,sizeof(ad));
    }
    void add(ll x)
    {
        ll u=0;
        for(int i=31;i>=0;i--)
        {
            int t=(x>>i)&1;
            if(!ch[u][t]) ch[u][t]=++sz;
            if(t)
            {
                ad[i]=(ad[i]+val[ch[u][t^1]])%mo;
            }
            else
            {
                ad[i]=(ad[i]-val[ch[u][t^1]]+mo)%mo;
                ans=(ans+val[ch[u][t^1]])%mo;
                // cout<<x<<" "<<i<<" "<<val[ch[u][t^1]]<<endl;
            }
            u=ch[u][t];
            val[u]++;
        }
    }
}T;
int main()
{
    T.init();
    scanf("%lld%lld",&n,&m);
    ans=0;
    for(int i=0;i<n;i++)
    {
        scanf("%lld",&a);
        T.add(a);
    }
    ans=ans*m%mo;
    //cout<<ans<<endl;
    ll pos=0;
    while(m>(1<<pos)) pos++;
    for(ll j=pos;j>=0;j--)
    {
        for(ll i=pos;i>=j;i--)
        {
        if(i==j)
        {
            if(m&(1<<i)) sum[j]+=(m%(1<<i))+1;
            else break;
        }
        else
        {
            if(m&(1<<i))
            {
                sum[j]+=(1<<(i-1));
            }
        }
        }
    }
    for(int i=0;i<=pos;i++)
    {
        ans=(ans+sum[i]*T.ad[i]%mo+mo)%mo;
    }
    printf("%lld\n",ans);
    return 0;
}

 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值