The Battle of Chibi---赤壁之战(DP+数据结构(树状数组))

题目:赤壁之战

一个长度为N的序列A,求A有多少个长度为M的严格递增子序列。

输入格式
  第一行包含整数T,表示共有T组测试数据。 每组数据,第一行包含两个整数N和M。 第二行包含N个整数,表示完整的序列A。

输出格式
  每组数据输出一个结果,每个结果占一行。输出格式为“Case #x: y”,x为数据组别序号,从1开始,y为结果。由于数据可能很大,请你输入对109+7109+7取模后的结果。

数据范围
  1≤T≤1001≤T≤100,
  1≤M≤N≤10001≤M≤N≤1000,
  序列中的整数的绝对值不超过109109。

输入样例:
2
3 2
1 2 3
3 2
3 2 1
输出样例:
Case #1: 3
Case #2: 0
思路:dp[i][j]表示长度为i,当前到 j 个数最长上升序列
状态转移:dp[i][j]+=dp[i][k] {j>k&&a[j]>a[k]}

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#define N 1001
using namespace std;
const long long mod=1000000007;
int a[N],dp[N][N],sum;
int main()
{
    int t;
    cin>>t;
    for(int s=1;s<=t;s++)
    {
        memset(dp,0,sizeof(dp));
        int n,m;
        cin>>n>>m;
        for(int i=1; i<=n; i++) cin>>a[i];
        a[0]=-(1<<30);
        dp[0][0]=1;
        for(int i=1; i<=m; i++)
        {
            for(int j=1; j<=n; j++)
            {
                for(int k=0; k<j; k++)
                {
                    if(a[j]>a[k])  dp[i][j] =(dp[i][j]+dp[i-1][k])%mod;
                }
            }
        }
        long long ans=0;
        for(int i=1; i<=n; i++)
        {
            ans=(ans+dp[m][i])%mod;
        }
        printf("Case #%d: %lld\n",s,ans);
    }
}

优化算法:因为每次j增长1,只是k的取值范围只是从1~j 到 1~j+1,所以可以用树状数组储存1到j中的可选值,然后j增长时,增加j+1范围。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#define N 1001
#define ll long long
using namespace std;
const long long mod=1000000007;
int a[N],b[N],n,m,dp[N][N],sum,c[N];
int lowbit(int x){
    return x&(-x);
}
ll ask(int x){
    ll ans=0;
    for(;x;x-=lowbit(x)){
        ans=(ans+c[x])%mod;
    }
    return ans;
}
void add(int x,int val){
    for(;x<=n;x+=lowbit(x)){
        c[x]=(c[x]+val)%mod;
    }
}
int main()
{
    int t;
    cin>>t;
    for(int s=1;s<=t;s++)
    {
        memset(dp,0,sizeof(dp));
        cin>>n>>m;
        for(int i=1; i<=n; i++) {
            cin>>a[i];
            b[i]=a[i];
        }
        sort(b+1,b+1+n);
        for(int i=1;i<=n;i++) {
            a[i]=lower_bound(b+1,b+1+n,a[i])-(b-1);//a[i]为在整个数组中的大小排名
        }// -----离散化,将a[i]将数字缩小,但仍保持相对大小关系
        dp[0][0]=1;
        for(int i=1; i<=m; i++)
        {
            memset(c,0,sizeof(c));
            if(i==1) add(1,dp[i-1][0]);
            for(int j=1; j<=n; j++)
            {
                dp[i][j]=ask(a[j]-1);
                add(a[j],dp[i-1][j]);
            }
        }
        long long ans=0;
        for(int i=1; i<=n; i++)
        {
            ans=(ans+dp[m][i])%mod;
        }
        printf("Case #%d: %lld\n",s,ans);
    }
}

  • 0
    点赞
  • 3
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值