Codeforce 218 div2

D

　　一开始想错了,试图用"前缀和-容量"来求从上层流下来了多少水",但这是错的,因为溢出可能发生在中间.

　　然后发现对于每层,溢出事件只会发生一次,所以可以用类似并查集的办法,在每次加水后直接更新每层的容量就行了.

E

　　推出公式后发现只要预处理坐标前缀和sum[]和sum的前缀和ssum[]就可以在o(1)时间内求出给定区间的答案,

　　接下来枚举区间就行了(区间长度是给定的):

　　左端点右移,需要先删除第一个点,相当于后面每个点产生的统计值减去一个值v,v可以利用sum快速计算;

　　添加右端点,需要加上一个值v,v可以利用sum和ssum快速计算.

　　

 struct node
 {
     int id;
     llong v;
 };node x[maxn];

 llong n,k,sum[maxn],ssum[maxn];
 llong getsum(int l,int r){return l==?sum[r]:sum[r]-sum[l-];}
 llong getssum(int l,int r)
 {
     if (l<=)
     {
         if (r<) return 0ll;
         return ssum[r];
     }else return ssum[r]-ssum[l-];
 }
 bool cmp(node u,node v) {return u.v<v.v;}
 int main()
 {
 //    freopen("test.txt","r",stdin);
     scanf("%I64d",&n);
     for (int i= ; i<=n ; i++ )
     {
         scanf("%I64d",&x[i].v);
         x[i].id = i;
     }
     sort(x+,x++n,cmp);
 //    for (int i=1 ; i<=n ; i++ ) printf("%I64d%c",x[i].v,i==n?'\n':' ');
     scanf("%I64d",&k);
     for (int i= ; i<=n ; i++ ) sum[i] = sum[i-]+x[i].v;
     for (int i= ; i<=n ; i++ ) ssum[i] = ssum[i-]+sum[i];
     llong f=;
     llong ans;
     int p=;
 //    printf("sum:\n");
 //    for (int i=1 ; i<=n ; i++ ) printf("%I64d%c",sum[i],i==n?'\n':' ');
 //    printf("ssum:\n");
 //    for (int i=1 ; i<=n ; i++ ) printf("%I64d%c",ssum[i],i==n?'\n':' ');
     for (int i= ; i<=n ; i++ )
     {
         if (i<k) f += (llong)(i+)*x[i].v;
         else
         {
             f -= getsum(i-k,i-);
             f += k*x[i].v;
             llong a = getssum(i-k+,i)-k*sum[i-k];
             llong b = getssum(i-k,i-)-k*sum[i-k];
             llong c = getsum(i-k+,i);
             llong tmp = 2ll*f-a-b-c;
     //        printf("i=%d f=%I64d a=%I64d b=%I64d c=%I64d tmp=%I64d\n",i,2ll*f,a,b,c,tmp);
             if (!p||ans>tmp)
                 ans = tmp,p=i;
         }
     }
     for (int i=p-k+ ; i<=p ; i++ ) printf("%d%c",x[i].id,i==p?'\n':' ');
     return ;
 }