程序師世界是廣大編程愛好者互助、分享、學習的平台,程序師世界有你更精彩!
首頁
編程語言
C語言|JAVA編程
Python編程
網頁編程
ASP編程|PHP編程
JSP編程
數據庫知識
MYSQL數據庫|SqlServer數據庫
Oracle數據庫|DB2數據庫
 程式師世界 >> 編程語言 >> C語言 >> C++ >> C++入門知識 >> Hdu 2430 Beans (數據結構_單調隊列)

Hdu 2430 Beans (數據結構_單調隊列)

編輯:C++入門知識

題目大意:有n坨豌豆,每坨都有w[i]個,現在要從中選擇連續的若干坨,然後用一個能裝p個豌豆的背包裝豆子,直到剩下的豌豆數量<=p為止,還要求剩下的豌豆數量少於k,問最多能裝多少包。

解題思路:題意很明了,模型也很好抽象。本題就是要選擇連續的一段和為sum,使得max(sum/p) 且sum%p<=k。那麼要怎麼算這個sum呢?這裡就大有文章可做了。
    其實這類求區間和的問題很經常是用sum[i]- sum[j-1]這種方式來表達的,我們設sum[i]表示0到i坨豌豆的總數量,那麼上面的sum就是sum[i]-sum[j-1]。從而得到一個式子,max(sum[i]-sum[j]) (j < i && (sum[i]-sum[j])%p<=k),表示以i為區間尾,因為豌豆數量非負數,那我們向前找個最小的下標使得差最小且符合條件即可。
    我們設sum[i]為x,sum[j]為y,那麼上式變成0<=(x-y)%p<=k且x-y最大,約束條件是x大等於y,我們要利用這些條件快速地找到x對應的那個最小的y。
    0<=(x-y)%p<=k  -->  0<=(x%p - y%p + p) % p<=k
    如果x%p >= y%p,那麼x%p-k<=y%p<=x%p,我們就可以根據x%p來定位y%p.
    如果x%p < y%p,那麼我們可以換個角度想,在以y結尾的時候就會有上面的情況。
    那問題就剩下一個了,如果根據x%p來快速定位y%p。寫法有多種,用線段樹和樹狀數組都可以寫,但我用線性的單調隊列來寫,時間78ms,是提交記錄的第二名。
    我先將每個和對p取余得到一個余數r數組,然後按余數大小優先位置其次的順序排序。排完序之後就可以根據ri - k <= rj <= ri來往前找那個最小的j。遍歷數組r,因為r是有序的,我用單調隊列儲存r小於當前的最小下標。關於單調隊列,我就不多說了,百度google去。最後更新答案即可。
    總復雜度還是(nlogn),主要用於排序。

測試數據:
20
2 19 3
17 3
Out:Case 1: 1

2 19 1
17 3
Out:Case 2: 1

2 19 3
17 6
Out:Case 3: -1

10 20 10
0 3 1 8 19 39 2 9 1 8
Out:Case 4: 4

3 100 10
32 34 23
Out:Case 5: -1

1 5 3
5
Out:Case 6: 1

5 15 4
7 8 3 10 10
Out:Case 7: 2

C艹代碼:
[cpp] 
#include <stdio.h> 
#include <string.h> 
#include <algorithm> 
using namespace std; 
#define MAX 1000010 
#define max(a,b) (a)>(b)?(a):(b) 
 
 
struct node { 
     
    int x,in; 
}r[MAX],qu[MAX]; 
__int64 sum[MAX],ans; 
int head,tail; 
int n,p,k,mmin[MAX]; 
 
 
int cmp(node a,node b) { 
     
    if (a.x == b.x) return a.in < b.in; 
    else return a.x < b.x; 

void Solve_AC() { 
     
    int i,j; 
    head = 0,tail = 1; 
 
 
    for (i = 1; i <= n; ++i) { 
         
        if (tail > head) mmin[r[i].in] = -1; 
        else mmin[r[i].in] = qu[tail].in; 
 
 
        while (tail <= head  
            && r[i].in < qu[head].in)  head--; 
 
 
        qu[++head] = r[i]; 
        while (tail <= head && r[i + 1].x - qu[tail].x > k) 
            tail++; 
    } 
     
     
    for (i = 1; i <= n; ++i) { 
         
        if (sum[i] % p <= k) ans = max(ans,sum[i] / p); 
        if (mmin[i] == -1 || mmin[i] >= i) continue; 
        ans = max(ans,(sum[i] - sum[mmin[i]]) / p); 
    } 

 
 
int main() 

    int i,j,t,cas = 0; 
     
     
    scanf("%d",&t); 
    while (t--) { 
         
        scanf("%d%d%d",&n,&p,&k); 
        for (i = 1; i <= n; ++i) { 
             
            scanf("%d",&sum[i]); 
            r[i].in = i; 
            sum[i] +=  sum[i-1]; 
            r[i].x = sum[i] % p; 
        } 
         
         
        sort(r+1,r+1+n,cmp); 
        ans = -1,Solve_AC(); 
        printf("Case %d: %I64d\n",++cas,ans); 
    } 

作者:woshi250hua

  1. 上一頁:
  2. 下一頁:
Copyright © 程式師世界 All Rights Reserved