当前位置: 首页 > 工具软件 > WMQ > 使用案例 >

北方大学 ACM多校训练 第六场-1007- wmq的队伍(树状数组+dp)

陈畅
2023-12-01

题目链接:wmq的队伍

dp[i][j] 表示第i个人且他是逆序序列的第j个时的方案数。
dp[i][j]=i1k=1dp[k][j1]
可以用树状数组来维护 ki=1dp[i][j]
最后可以滚动数组优化一下dp

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int maxn=2e4+7;
const ll mod=1e9+7;
ll c[107][maxn],dp[107];
int a[maxn],n,k;
int lowbit(int x)
{
    return x&-x;
}
void modify(int k, int x, ll val)
{
    for(;x<maxn;x+=lowbit(x))
        c[k][x]=(c[k][x]+val)%mod;
}
ll query(int k,int x)
{
    ll res=0;
    for(;x>0;x-=lowbit(x))
        res=(c[k][x]+res)%mod;
    return res;
}
int main()
{
    ios::sync_with_stdio(false);
    int T;
    cin>>T;
    while(T--)
    {
        cin>>n>>k;
        for(int i=1;i<=n;i++)
            cin>>a[i];
        memset(dp,0,sizeof(dp));
        memset(c,0,sizeof(c));
        ll ans=0;
        modify(0,maxn-1,1);
        for(int i=1;i<=n;i++)
        {
            for(int j=k;j>=1;j--)
            {
                dp[j]=(query(j-1,maxn-1)-query(j-1,a[i])+mod)%mod;
                modify(j,a[i],dp[j]);
            }
            ans=(ans+dp[k])%mod;
        }
        cout << ans << endl;
    }
    return 0;
}
 类似资料: