CCF CSP 202109-2 非零段划分

CCF CSP 202109-2 非零段划分

试题编号： 202109-2
试题名称： 非零段划分
时间限制： 1.0s
内存限制： 512.0MB

题目描述

输入格式

输出格式

样例1输入:

11
3 1 2 0 0 2 0 4 5 0 2

样例1输出：

5

样例1解释：
p=2时，A=[3,0,2,0,0,2,0,4,5,0,2]，5个非零段依次为[3]、[2]、[2]、[4,5] 和 [2],此时非零段个数达到最大。

样例2输入:

14
5 1 20 10 10 10 10 15 10 20 1 5 10 15

样例2输出：

4

样例2解释：
p=12时，A=[0,0,20,0,0,0,0,15,0,20,0,0,0,15]， 4个非零段依次为 [20]、[15]、[20] 和 [15]，此时非零段个数达到最大。

样例3输入:

3
1 0 0

样例3输出：

1

样例3解释：
p=1时，A=[1,0,0]，此时仅有1个非零段 [1]，非零段个数达到最大。

样例4输入:

3
0 0 0

样例4输出：

0

样例4解释：
无论p取何值，A都不含有非零段，故非零段个数至多为0。

解题思路：

直接暴力找p肯定会超时，但是p的大小并不是我们需要关注的，我们只用找到最大非零段的个数是多少就行。
因为小于p的值都会变为0，容易想到某个数假设为0，那么小于这个数的所有数都会变为0。
因此，我们对数组A进行结构体排序，按照数值大小进行排序，并记录下来数的位置，从小的数开始变为0。那么，如果某个数的左右位置都没有被删除（vis记录为0）过，那么删除当前非0段数为加1。如果某个数的左右位置都被删除（vis记录为1）过，那么非0段会减1。其余情况，非0段个数不变。记录更新答案就可。
注意：由于小于p的值都会变成0，所以在更新ans时候，必须当前数不等于now才能更新答案

代码如下：

#include <bits/stdc++.h>

using namespace std;

const int N=5e5+5;
struct node {
    int x;
    int id;
}a[N];

bool cmp(node A,node B) {
    if(A.x==B.x)
        return A.id<B.id;
    return A.x<B.x;
}

int main()
{
    ios::sync_with_stdio(false);
    int n;
    cin>>n;
    for(int i=1;i<=n;++i) {
        cin>>a[i].x;
        a[i].id=i;
    }
    sort(a+1,a+1+n,cmp);

    int vis[N]={0};
    vis[0]=1,vis[n+1]=1;
    int ans=0,now=-1;
    int cnt=1;  //当前的非0区间数为1

    for(int i=1;i<=n;++i) {
        if(a[i].x!=0&&a[i].x!=now)    //因为全是0是不用更新的并且小于P相同的数都要变为0
            ans=max(ans,cnt);
        int id=a[i].id;
        vis[id]=1;
        if(vis[id-1]!=1&&vis[id+1]!=1)  //左右两个位置都没被访问过
            ++cnt;
        else if(vis[id-1]==1&&vis[id+1]==1) //左右两个位置都被访问过
            --cnt;
        now=a[i].x;
    }
    cout<<ans<<endl;
    return 0;
}