HNOI2002 营业额统计 (sbt)

1588: [HNOI2002]营业额统计

Time Limit: 5 Sec   Memory Limit: 162 MB
Submit: 11740   Solved: 4178
[ Submit][ Status][ Discuss]

Description

营业额统计 Tiger最近被公司升任为营业部经理，他上任后接受公司交给的第一项任务便是统计并分析公司成立以来的营业情况。 Tiger拿出了公司的账本，账本上记录了公司成立以来每天的营业额。分析营业情况是一项相当复杂的工作。由于节假日，大减价或者是其他情况的时候，营业额会出现一定的波动，当然一定的波动是能够接受的，但是在某些时候营业额突变得很高或是很低，这就证明公司此时的经营状况出现了问题。经济管理学上定义了一种最小波动值来衡量这种情况： 该天的最小波动值 当最小波动值越大时，就说明营业情况越不稳定。 而分析整个公司的从成立到现在营业情况是否稳定，只需要把每一天的最小波动值加起来就可以了。你的任务就是编写一个程序帮助Tiger来计算这一个值。 第一天的最小波动值为第一天的营业额。  输入输出要求

Input

第一行为正整数 ，表示该公司从成立一直到现在的天数，接下来的n行每行有一个整数(有可能有负数) ，表示第i天公司的营业额。

Output

输出文件仅有一个正整数，即Sigma(每天最小的波动值) 。结果小于2^31 。

Sample Input

6
5
1
2
5
4
6

Sample Output

12

HINT

结果说明：5+|1-5|+|2-1|+|5-5|+|4-5|+|6-5|=5+4+1+0+1+1=12

此题数据有问题，详见讨论版http://www.lydsy.com/JudgeOnline/wttl/wttl.php?pid=1588

Source

评测链接：http://www.lydsy.com/JudgeOnline/problem.php?id=1588

                              http://tyvj.cn/p/1185

解法：SBT。

           对于第i天的营业额ai，它的最小波动值m就等于从1号到i-1天中，与它大小最接近的营业额p与ai差值的绝对值。那么由此可以想到，在将ai插入sbt树时，所经过的节点中，必定有一个就是就是我们要找的p号点，也就得到第i天的最小波动值m=min{m,abs(p[t]-a[i])}，p[t]是ai插入sbt树时途经的点t的营业额值。

由此基本思路就出来了，维护一棵sbt树，每次插入时维护出最小波动值m，然后将每次的m累加起来即可。

         还要注意，本题的数据存在问题：

                 ①测试数据中存在负数与0，并不全是正整数。

                 ②lydsy的数据少了一行，少的这一个数据当成0来处理。所以你如果用手写读入优化，但是没有判断eof的话，就会re。

oj能评测通过的代码：

#include<cstdio>
#include<cstring>
#include<cctype>
#include<cmath>
#include<algorithm>
using namespace std;

const int maxn=32767;
int s[maxn+10],l[maxn+10],r[maxn+10],p[maxn+10];

void left_rotate(int &t)
{
  int k=r[t]; r[t]=l[k],l[k]=t;
  s[k]=s[t],s[t]=s[l[t]]+s[r[t]]+1,t=k;
}

void right_rotate(int &t)
{
  int k=l[t]; l[t]=r[k],r[k]=t;
  s[k]=s[t],s[t]=s[l[t]]+s[r[t]]+1,t=k;
}

void maintain(int &t,bool flag)
{
  if(flag)
    if(s[l[l[t]]]>s[r[t]])right_rotate(t);
    else 
      if(s[r[l[t]]]>s[r[t]])left_rotate(l[t]),right_rotate(t);
      else return;
  else
    if(s[r[r[t]]]>s[l[t]])left_rotate(t);
    else
      if(s[l[r[t]]]>s[l[t]])right_rotate(r[t]),left_rotate(t);
      else return;
  maintain(l[t],1);
  maintain(r[t],0);
  maintain(t,1);
  maintain(t,0);
}


void insert(int &t,int k,int &w)
{
  if(t==0){t=k;return;}
  s[t]++,w=min(w,abs(p[t]-p[k]));
  if(p[k]<p[t])insert(l[t],k,w);
  else insert(r[t],k,w);
  maintain(t,p[k]<p[t]);
}

int main()
{
  //freopen("turnover.in","r",stdin);
  //freopen("turnover.out","w",stdout);
  
  int n,ans,root=1,i,k,w;
  scanf("%d%d",&n,&ans);
  s[1]=1,p[1]=ans;
  for(i=2;i<=n;i++)
    {
      int j=0;scanf("%d",&j);//这里的j一定要赋初值为0，因为数据少了一行，要当成0来处理的。
      p[(k=s[root]+1)]=j,s[k]=1;
      w=1e9,insert(root,k,w),ans+=w;
	}
  printf("%d\n",ans);
  return 0;
}

自己写的手写读入优化，因为数据问题，lydsy无法通过：

#include<cstdio>
#include<cstring>
#include<cctype>
#include<cmath>
#include<algorithm>
using namespace std;

const int maxn=32767;
int s[maxn+10],l[maxn+10],r[maxn+10],p[maxn+10];

int getin()
{
  int ans=0;char tmp;bool sign=0;
  while(!isdigit(tmp=getchar()) && tmp!='-');
  if(tmp=='-')sign=1,tmp=getchar();
  do ans=(ans<<3)+(ans<<1)+tmp-'0';
  while(isdigit(tmp=getchar()));
  return sign?-ans:ans;
}

void left_rotate(int &t)
{
  int k=r[t]; r[t]=l[k],l[k]=t;
  s[k]=s[t],s[t]=s[l[t]]+s[r[t]]+1,t=k;
}

void right_rotate(int &t)
{
  int k=l[t]; l[t]=r[k],r[k]=t;
  s[k]=s[t],s[t]=s[l[t]]+s[r[t]]+1,t=k;
}

void maintain(int &t,bool flag)
{
  if(flag)
    if(s[l[l[t]]]>s[r[t]])right_rotate(t);
    else 
      if(s[r[l[t]]]>s[r[t]])left_rotate(l[t]),right_rotate(t);
      else return;
  else
    if(s[r[r[t]]]>s[l[t]])left_rotate(t);
    else
      if(s[l[r[t]]]>s[l[t]])right_rotate(r[t]),left_rotate(t);
      else return;
  maintain(l[t],1);
  maintain(r[t],0);
  maintain(t,1);
  maintain(t,0);
}


void insert(int &t,int k,int &w)
{
  if(t==0){t=k;return;}
  s[t]++,w=min(w,abs(p[t]-p[k]));
  if(p[k]<p[t])insert(l[t],k,w);
  else insert(r[t],k,w);
  maintain(t,p[k]<p[t]);
}

int main()
{
  int n=getin(),ans=getin(),root=1,i,k,w;
  s[1]=1,p[1]=ans;
  for(i=2;i<=n;i++)
    {
      p[(k=s[root]+1)]=getin(),s[k]=1;
      w=1e9,insert(root,k,w),ans+=w;
	}
  printf("%d\n",ans);
  return 0;
}