求lcp : Trie + lca
题目:
https://www.luogu.org/problem/show?pid=1311
挺有意思的一道题;
先考虑n^2*k的做法;
开链表直接模拟;
用st表维护区间最小值;
pre[i]:与i颜色相同的前一个客栈的位置;
h[x]:最后一个颜色为x的客栈的位置;
#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
using namespace std;
const int MAXN=200001;
int dp[MAXN][21],pre[MAXN],h[MAXN];
int n,k,c,tot;
struct ST
{
void st_init()
{
for(int j=1;(1<<j)<=n;j++)
for(int i=1;i+(1<<j)-1<=n;i++)
dp[i][j]=min(dp[i][j-1],dp[i+(1<<(j-1))][j-1]);
return;
}
int query(int l,int r)
{
int k=log2(r-l+1);
return min(dp[l][k],dp[r-(1<<k)+1][k]);
}
}st;
void solve()
{
int x;
memset(h,0,sizeof(h));
memset(pre,0,sizeof(pre));
scanf("%d%d%d",&n,&k,&c);
for(int i=1;i<=n;i++)
{
scanf("%d%d",&x,&dp[i][0]);
pre[i]=h[x];
h[x]=i;
}
st.st_init();
for(int x=0;x<k;x++)
for(int i=h[x];i;i=pre[i])
for(int j=pre[i];j;j=pre[j])
{
int xx=j,yy=i;
if(st.query(xx,yy)<=c) tot++;
}
cout<<tot;
}
int main()
{
solve();
return 0;
}
我们发现,在找相同颜色客栈的方案数时,有许多的方案数被重复计算;
所以,考虑优化;
tot[k]:记录之前颜色为k的客栈出现的次数;
sum[k] : 记录之前“已找到小于花费”的颜色为k的客栈出现次数;
h[k] : 上一个颜色为k的客栈出现的位置;
所以两个颜色相同的客栈之间有小于花费的咖啡店;
将sum更新为tot即可;
ST表:O(nlogn)
模拟:O(n);
总复杂度O(nlogn);
#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
using namespace std;
const int MAXN=200001;
int dp[MAXN][21],pre[MAXN],h[MAXN],cnt[MAXN];
int n,k,c,tot[MAXN],sum[MAXN],s[MAXN],ans;
struct ST
{
void st_init()
{
for(int j=1;(1<<j)<=n;j++)
for(int i=1;i+(1<<j)-1<=n;i++)
dp[i][j]=min(dp[i][j-1],dp[i+(1<<(j-1))][j-1]);
return;
}
int query(int l,int r)
{
int k=log2(r-l+1);
return min(dp[l][k],dp[r-(1<<k)+1][k]);
}
}st;
void solve()
{
int x;
memset(h,0,sizeof(h));
memset(pre,0,sizeof(pre));
scanf("%d%d%d",&n,&k,&c);
for(int i=1;i<=n;i++)
scanf("%d%d",&s[i],&dp[i][0]);
st.st_init();
for(int i=1;i<=n;i++)
{
if(!h[s[i]])
tot[s[i]]++,h[s[i]]=i;
else
{
if(st.query(h[s[i]],i)<=c)
ans+=tot[s[i]],sum[s[i]]=tot[s[i]];
else ans+=sum[s[i]];
tot[s[i]]++,h[s[i]]=i;
}
}
cout<<ans;
}
int main()
{
solve();
return 0;
}
其实ST表也可以优化掉;
我们只需要找到离当前“最近”(也可以是当前位置)的小于花费的客栈;
如果这个客栈在上一个与当前颜色相同的客栈后面,那么直接更新sum,sum=tot;
此时的时间复杂度为O(n);
#include<cstdio>
int n,k,c,now,x,y,ans;
int sum[1001],tot[1001],h[1001];
int main()
{
scanf("%d%d%d",&n,&k,&c);
for(int i=1;i<=n;i++)
{
scanf("%d%d",&x,&y);
now=(y<=c?i:now);
if(now>=h[x]) sum[x]=tot[x];
h[x]=i,ans+=sum[x],tot[x]++;
}
printf("%d",ans);
return 0;
}
总结:
1.这种题还是要多练;
2.考虑用数据结构进行优化;