https://codeforces.com/group/NVaJtLaLjS/contest/253990/attachments
思路:
设
l
e
n
(
i
,
j
)
:
len(i,j):
len(i,j):以
i
i
i为结尾,长度为j的合法子序列个数。那么
f
(
i
,
j
)
=
∑
k
=
1
i
−
1
f
(
k
,
j
−
1
)
f(i,j)=\sum_{k=1}^{i-1}f(k,j-1)
f(i,j)=∑k=1i−1f(k,j−1),用树状数组优化。(
n
3
−
>
n
2
∗
l
o
g
(
n
)
n^3 -> n^2*log(n)
n3−>n2∗log(n))注意树状数组的sum函数里面要加取模!
l
e
n
s
u
m
(
i
)
为
长
度
为
i
的
合
法
子
序
列
个
数
lensum(i)为长度为i的合法子序列个数
lensum(i)为长度为i的合法子序列个数
设
c
n
t
(
i
)
:
i
cnt(i):i
cnt(i):i个不同的字母拼成n个的组合数目。那么由容斥原理得
c
n
t
(
i
)
=
i
n
−
∑
j
=
1
i
−
1
C
i
j
∗
c
n
t
(
j
)
cnt(i)=i^n-\sum_{j=1}^{i-1}C_i^j*cnt(j)
cnt(i)=in−∑j=1i−1Cij∗cnt(j)
答案就是
∑
i
=
1
n
l
e
n
s
u
m
(
i
)
∗
c
n
t
(
i
)
\sum_{i=1}^n lensum(i)*cnt(i)
∑i=1nlensum(i)∗cnt(i)
#include<bits/stdc++.h>
using namespace std;
#define ll long long
#define lowbit(x) (x&(-x))
const int maxn=2050;
const int mod=1000000007;
ll n,a[maxn],len[maxn][maxn],lensum[maxn],c[maxn],cnt[maxn],C[maxn][maxn];
int sum(int x)
{
ll ret=0;
while(x)ret=(ret+c[x])%mod,x-=lowbit(x);
return ret;
}
void add(int x,int d)
{
while(x<=n)c[x]+=d,x+=lowbit(x);
}
ll pow_mod(ll a,ll n)
{
if(!n)return 1;
ll x=pow_mod(a,n/2);
x=x*x%mod;
if(n&1)x=x*a%mod;
return x;
}
int main()
{
//freopen("input.in","r",stdin);
cin>>n;
for(int i=0;i<=n;i++)
{
C[i][0]=1;
for(int j=1;j<=i;j++)C[i][j]=(C[i-1][j-1]+C[i-1][j])%mod;
}
for(int i=1;i<=n;i++)cin>>a[i];
for(int i=1;i<=n;i++)len[a[i]][1]=1;
for(int j=2;j<=n;j++)
{
for(int i=1;i<=n;i++)
{
len[a[i]][j]=sum(a[i]-1);
add(a[i],len[a[i]][j-1]);
}
memset(c,0,sizeof(c));
}
for(int j=1;j<=n;j++)for(int i=1;i<=n;i++)lensum[j]=(lensum[j]+len[i][j])%mod;
for(int i=1;i<=n;i++)
{
cnt[i]=pow_mod(i,n);
for(int j=i-1;j>0;j--)cnt[i]=(cnt[i]-cnt[j]*C[i][j]%mod+mod)%mod;
}
ll ans=0;
for(int i=1;i<=n;i++)ans=(ans+cnt[i]*lensum[i])%mod;
cout<<ans<<endl;
return 0;
}