[USACO Jan08] 化装晚会

万圣节又到了！Farmer John打算带他的奶牛去参加一个化装晚会，但是，FJ只做了一套能容 下两头总长不超过S(1 <= S <= 1,000,000)的牛的恐怖服装。FJ养了N(2 <= N <= 20,000)头按1..N顺序编号的奶牛，编号为i的奶牛的长度为L_i(1 <= L_i <= 1,000,000)。如果两头奶牛的总长度不超过S，那么她们就能穿下这套服装。

FJ想知道，如果他想选择两头不同的奶牛来穿这套衣服，一共有多少种满足条件的方案。

程序名: costume

输入格式:

• 第1行: 2个用空格隔开的整数：N 和 S

• 第2..N+1行: 第i+1为1个整数：L_i

输入样例 (costume.in):

4 6
3
5
2
1

输出格式:

• 第1行: 输出1个整数，表示FJ可选择的所有方案数。注意奶牛顺序不同的两种方案是被视为相同的

输出样例 (costume.out):

4

输出说明:

4种选择分别为：奶牛1和奶牛3；奶牛1和奶牛4；奶牛2和奶牛4；奶牛3和奶牛4。

 

 

简单的单调队列。

#include<iostream>
#include<fstream>
using namespace std;
ifstream fin("costume.in");
ofstream fout("costume.out");

int n,s,h[20005],tot[20005]={0};

void Quick(int l,int r){
     if(l>=r) return ;
     int i=l,j=r,mid=h[(l+r)/2];
     while(i<=j)
     {
       while(h[i]<mid) i++;
       while(h[j]>mid) j--;
       if(i<=j) {swap(h[i],h[j]);i++;j--;}
                }
     Quick(l,j);
     Quick(i,r);
     }

int main()
{
    fin>>n>>s;
    
    for(int i=1;i<=n;++i) fin>>h[i];
    
    Quick(1,n);
    
    int l=1,r=2;
    while(h[l]+h[r]<=s&&r<=n) tot[l]++,r++;
    r--;
    l=2;
    while(l<r&&r<=n)
    {
      tot[l]+=tot[l-1]-1;
      while(h[l]+h[r]>s&&r>l) {tot[l]--;r--;}
      l++;
               }
    
    int ans=0;
    for(int i=1;i<=n;++i)
    ans+=tot[i];
    
    fout<<ans<<endl;
    
    return 0;
    
    }