1. 程式人生 > >Codeforces Round 52.C+求特殊陣列

Codeforces Round 52.C+求特殊陣列

題目連結:http://codeforces.com/contest/1065/problem/C
題目大意:
在這裡插入圖片描述
5個方柱,高度分別為3,1,2,2,4。要把他們切的一樣平,且每次切的方塊數量不超過5,如圖切2刀y=2, y=1。

在這裡插入圖片描述
思路:只要把方塊高度排個序,再求一個數組s[],s[i]表示y=i到y=i+1這層有多少個方塊。
如果對於每個i計算都遍歷一遍高度時間複雜度為O(n^2)。
這裡用了一種O(n)的方法

陣列求法:

int n, k;
    scanf("%d%d",&n,&k);
    for(int i=1;i<=n;i++)
    {
        scanf("%d",&a[i]);
        s[a[i]-1]++;
    }
    sort(a+1, a+n+1);

    for(int i=a[n]-1;i>=0;i--)
    {
        s[i]+=s[i+1];
    }

這樣我們只要再模擬一遍就行了。

AC程式碼

#include<bits/stdc++.h>
using namespace std;

int a[200005];
int s[200005];

int main()
{
    int n, k;
    scanf("%d%d",&n,&k);
    for(int i=1;i<=n;i++)
    {
        scanf("%d",&a[i]);
        s[a[i]-1]++;
    }
    sort(a+1, a+n+1);

    for(int i=a[n]-1;i>=0;i--)
    {
        s[i]+=s[i+1];
    }

    int sum=0, now=0;
    for(int i=a[n]; i>=1; i--)
    {
        if(now+s[i]>k)
            sum++, now=s[i];
        else
            now+=s[i];

        if(i==a[1])//最小高度不用再切了
        {
            if(now!=0)//處理剩餘方塊,加now!=0判斷防止全部方塊一樣高
            sum++;
            break;
        }
    }

    cout<<sum<<endl;

    return 0;
}