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;
}