Codeforces Round 52.C+求特殊陣列
阿新 • • 發佈:2018-11-17
題目連結: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; }