题意:
n天,每天得到一个值,要求输出每一天和这天得到的值相差最小的之前天得到的值与这个值的差的和。n不知道,不过O(nlog2n)可写。
题解:
说是平衡树模板题,不过可以用set水过去。先在set插入一个-INF和INF防溢出(yyl大爷教我的)每次在set中lower_bound这天得到的值,求出的是≤它的最大值,比较一下这个值和set中这个值的后继与这天得到的值的差。
代码:
1 #include <cstdio> 2 #include <cstring> 3 #include <algorithm> 4 #include <set> 5 #define INF 2147483647 6 #define inc(i,j,k) for(int i=j;i<=k;i++) 7 using namespace std; 8 9 int n,ans,x; set <int> st; 10 int main(){ 11 scanf("%d",&n); st.clear(); st.insert(INF); st.insert(-INF); ans=0; 12 inc(i,1,n){ 13 scanf("%d",&x); 14 if(i==1)ans+=x;else{ 15 int y=* (st.lower_bound(x)),z=* (--st.lower_bound(x)); //printf("%d %d\n",y,z); 16 if(y==INF)ans+=abs(x-z);else if(z==-INF)ans+=abs(x-y);else ans+=min(abs(x-z),abs(x-y)); 17 } 18 st.insert(x); 19 } 20 printf("%d",ans); return 0; 21 }
20160531