题意:对于可重数组 a a a的每个排列方式(方式不同当且仅当有一个位置的数不同)求逆序对数 t t t,求对于所有的 t t t的 b t b^t bt的和。
题解:考虑从小到大插入,先不考虑相等的情况,那么插入第i大的数时按插入位置的不同会分别增加
0
,
1
,
2
,
3
,
.
.
.
i
−
1
0,1,2,3,...i-1
0,1,2,3,...i−1个逆序对,就相当于答案
∗
=
(
1
+
b
+
b
2
+
b
3
.
.
.
b
i
−
1
)
*=(1+b+b^2+b^3...b^{i-1})
∗=(1+b+b2+b3...bi−1)。
相等时,我们考虑先看做不等然后去重,去重时,因为看做不相等时的方案统计已经限定了这些数的位置,那么实际上若有
k
k
k个相等的数,我们实际上需要去重的时候只需要考虑这些相等的数的相对位置而并不需要关心某个数后面还有几个空位可以插入。那么插入第i个相等的数的时按插入位置的不同会分别等方案数多算
0
,
1
,
2
,
3
,
.
.
.
i
−
1
0,1,2,3,...i-1
0,1,2,3,...i−1个逆序对,就相当于答案
/
=
(
1
+
b
+
b
2
+
b
3
.
.
.
b
i
−
1
)
/=(1+b+b^2+b^3...b^{i-1})
/=(1+b+b2+b3...bi−1)。
完了。
一定要注意计算方案时DP式子还限定了那些信息,这样才不会算重。
AC Code:
#include<bits/stdc++.h>
#define maxn 100005
#define mod 1000000007
using namespace std;
int n,b;
int pw[maxn],pws[maxn],pwsm[maxn];
map<int,int>mp;
int Pow(int base,int k){
int ret=1;
for(;k;k>>=1,base=1ll*base*base%mod)
if(k&1)
ret=1ll*ret*base%mod;
return ret;
}
int main(){
scanf("%d%d",&n,&b);
pw[0] = pws[0] = pwsm[0] = 1;
for(int i=1;i<=n;i++)
pw[i] = pw[i-1] * 1ll * b % mod,
pws[i] = (pws[i-1] + pw[i])%mod,
pwsm[i] = pwsm[i-1] * 1ll * pws[i] % mod;
int ans = pwsm[n - 1];
for(int i=1;i<=n;i++){
int a;scanf("%d",&a);
mp[a]++;
}
for(auto u:mp)
if(u.second > 1)
ans = 1ll * ans * Pow(pwsm[u.second-1],mod-2) % mod;
printf("%d\n",(ans+mod)%mod);
}