[洛谷]P1311 选择客栈

原题

---

首先暴力如果写的优秀,可以拿到60分,这里介绍两种暴力:

40:
#include<stdio.h>
#include<stdlib.h>
#include<math.h>
#include<string.h>
#include<algorithm>
using namespace std;
int color[200010],cost[200010];
int main(){
    int i,j,k,n,m,p,ans=0;
    scanf("%d%d%d",&n,&k,&p);
    for(i=1;i<=n;i++)
        scanf("%d%d",&color[i],&cost[i]);
    for(int o=1;o<=k;o++)
        for(i=1;i<=n;i++)
            for(j=1;j<=n;j++){
                int flag=0;
                for(m=i;m<=j;m++)
                    if(color[i]==color[j] && i!=j && cost[m]<=p)
                        flag=1;
                if(flag)ans++;
            }
    printf("%d\n",ans/k);
    return 0;
}
这是一种莫名其妙的卡常数,大家不要在意,我只是想拿30分而已......(可是还是没有30分!!!)

60:
#include<stdio.h>
#include<stdlib.h>
#include<math.h>
#include<string.h>
#include<algorithm>
using namespace std;
int color[200010],cost[200010];
int main(){
    int i,j,k,n,m,p,ans=0;
    scanf("%d%d%d",&n,&k,&p);
    for(i=1;i<=n;i++)
        scanf("%d%d",&color[i],&cost[i]);
    for(i=1;i<n;i++)
        for(j=i+1;j<=n;j++)
            if(color[i]==color[j]){
                int flag=0;
                for(m=i;m<=j;m++)
                    if(cost[m]<=p)flag=1;
                if(flag==1)ans++;
            }
    printf("%d\n",ans);
    return 0;
}

那么接下来看正解:
yzk大佬说过:

这就是一道简单的数论题.注意到颜色只有100种，多种颜色混合起来使问题变得更复杂，我们考虑单独考虑每一种颜色的情况

题目的含义为取两个颜色相同的客栈l,r，使得区间[l,r][l,r][l,r]中含有一个消费<=p的咖啡店。那么我们只需要留下有用的数据，若该咖啡店消费<=p则为true，>p则为false。若单独考虑只有一种颜色，就变成了有多少个区间[l,r][l,r][l,r]使得区间中含有至少一个true。

假设我们不管这个条件，那么该种颜色下的客栈个数为n时，我们可以选择的区间个数是为n*(n-1)/2的。

那我们如何统计含有true的区间个数呢？

反着统计，我们将所有的区间个数减去不含true的区间个数。

我们可以用一个数组ca保存符合条件的咖啡店的下标。

枚举相邻的两个咖啡店下标分别为a1,a2

则统计(a1,a2)(a1,a2)(a1,a2)这个区间里面有k个这种颜色的店子，他们的组合区间数为k*(k-1)/2，再将当前颜色的总组合数n*(n-1)/2减去这个数。因为(a1,a2)(a1,a2)(a1,a2)这个区间绝对不含满足条件的咖啡店。

每种颜色统计过后，加起来就是答案。

区间同颜色店子的个数可以通过前缀和得到。

每种颜色分别处理会比较好。

#include<bits/stdc++.h>
using namespace std;
int color[200010],a[200010],sum[200010],reans;
int main(){
    int can=0;
    int i,j,k,n,m,p;
    scanf("%d%d%d",&n,&k,&p);
    for(i=1;i<=n;i++){
        int c;
        scanf("%d%d",&color[i],&c);
        if(c<=p)a[++can]=i;
    }
    a[++can]=n+1;
    int ans=0;
    for(i=0;i<k;i++){
        ans=0;
        memset(sum,0,sizeof(sum));
        for(j=1;j<=n;j++){
            sum[j]=sum[j-1];
            if(color[j]==i)sum[j]++,ans++;
        }
        ans=(ans-1)*ans/2;
        int ans2=0;
        for(j=1;j<=can;j++){
            ans2=sum[a[j]-1]-sum[a[j-1]];
            ans-=ans2*(ans2-1)/2;
        }
        reans+=ans;
    }
    printf("%d\n",reans);
    return 0;
}