POJ 1952 BUY LOW, BUY LOWER

时间:2022-08-11 16:24:08

$dp$。

一开始想了一个$dp$做法,$dp[i][j]$表示前$i$个数字,下降序列长度为$j$的方案数为$dp[i][j]$,这样做需要先离散化然后用树状数组优化,空间复杂度为${n^2}$,时间复杂度为$O({n^2}\log n)$,这样的做法被$POJ$卡了内存。既然是$MLE$,然后我去$discuss$测了一下数据,发现答案都是对的。

$MLE$:

#pragma comment(linker, "/STACK:1024000000,1024000000")
#include<cstdio>
#include<cstring>
#include<cmath>
#include<algorithm>
#include<vector>
#include<map>
#include<set>
#include<queue>
#include<stack>
#include<iostream>
using namespace std;
typedef long long LL;
const double pi=acos(-1.0),eps=1e-;
void File()
{
freopen("D:\\in.txt","r",stdin);
freopen("D:\\out.txt","w",stdout);
} const int maxn=;
int f[maxn][maxn],h[maxn][maxn],n,a[maxn],b[maxn];
int c[maxn][maxn];
int pre[(<<)+]; int get(int x)
{
int L=,R=n,res;
while(L<=R)
{
int mid=(L+R)/;
if(b[mid]<x) L=mid+;
else if(b[mid]==x) res=mid,L=mid+;
else R=mid-;
}
return res;
} int lowbit(int x){return x&(-x);} int sum(int p,int x)
{
int res=;
for(int i=x;i>;i=i-lowbit(i)) res=res+c[p][i];
return res;
} void update(int p,int x,int val)
{
for(int i=x;i<=;i=i+lowbit(i)) c[p][i]=c[p][i]+val;
} int main()
{
scanf("%d",&n);
for(int i=;i<=n;i++) scanf("%d",&a[i]),b[i]=a[i];
sort(b+,b++n); for(int i=;i<=n;i++) a[i]=get(a[i]); memset(h,,sizeof h); memset(c,,sizeof c);
memset(f,,sizeof f);
memset(pre,,sizeof pre); h[][]=; f[][]=; update(,,); for(int i=;i<=n;i++)
{
for(int j=;j<=i;j++) h[i][j]=sum(j-,)-sum(j-,a[i]);
int p=pre[a[i]];
for(int j=;j<=n;j++) f[i][j]=h[i][j]-h[p][j];
for(int j=;j<=n;j++) update(j,a[i],f[i][j]);
pre[a[i]]=i;
}
for(int j=n;j>=;j--)
{
int ans=;
for(int i=;i<=n;i++) ans=ans+f[i][j];
if(ans==) continue;
else
{
printf("%d %d\n",j,ans);
break;
}
} return ;
}

事实上,上述做法中很多信息都是冗余的,我们只需记录到$i$位置的最长下降序列的长度$f[i]$以及方案数$g[i]$就可以了。

#pragma comment(linker, "/STACK:1024000000,1024000000")
#include<cstdio>
#include<cstring>
#include<cmath>
#include<algorithm>
#include<vector>
#include<map>
#include<set>
#include<queue>
#include<stack>
#include<iostream>
using namespace std;
typedef long long LL;
const double pi=acos(-1.0),eps=1e-;
void File()
{
freopen("D:\\in.txt","r",stdin);
freopen("D:\\out.txt","w",stdout);
} const int maxn=;
int n,a[maxn],f[maxn],g[maxn]; int main()
{
scanf("%d",&n);
for(int i=;i<=n;i++) scanf("%d",&a[i]); for(int i=;i<=n;i++)
{
f[i]=; g[i]=; bool flag=;
for(int j=i-;j>=;j--)
{
if(a[j]<a[i]) continue;
if(a[j]==a[i])
{
if(flag==) g[i]=;
break;
}
else if(a[j]>a[i])
{
flag=;
if(f[j]+>f[i]) f[i]=f[j]+,g[i]=g[j];
else if(f[j]+==f[i]) g[i]=g[i]+g[j];
}
}
} int ans=; for(int i=;i<=n;i++) ans=max(ans,f[i]);
int ans2=; for(int i=;i<=n;i++) if(f[i]==ans) ans2=ans2+g[i];
printf("%d %d\n",ans,ans2); return ;
}