本文主要是介绍【FZU】2171 防守阵地 II 线段树,希望对大家解决编程问题提供一定的参考价值,需要的开发者们随着小编来一起学习吧!
Problem 2171 防守阵地 IIAccept: 96 Submit: 360
Time Limit: 3000 mSec Memory Limit : 32768 KB
Problem Description
部队中总共有N个士兵,每个士兵有各自的能力指数Xi,在一次演练中,指挥部确定了M个需要防守的地点,指挥部将选择M个士兵依次进入指定地点进行防守任务,获得的参考指数即为M个士兵的能力之和。随着时间的推移,指挥部将下达Q个指令来替换M个进行防守的士兵们,每个参加完防守任务的士兵由于疲惫等原因能力指数将下降1。现在士兵们排成一排,请你计算出每次进行防守的士兵的参考指数。
Input
输入包含多组数据。
输入第一行有两个整数N,M,Q(1<=N<=100000,1<=M<=1000,1<=Q<=100000),第二行N个整数表示每个士兵对应的能力指数Xi(1<=Xi<=1000)。
接下来Q行,每行一个整数X,表示在原始队列中以X为起始的M个士兵替换之前的士兵进行防守。(1<=X<=N-M+1)
对于30%的数据1<=M,N,Q<=1000。
Output
输出Q行,每行一个整数,为每次指令执行之后进行防守的士兵参考指数。
Sample Input
2 1 3 1 4
1
2
3
Sample Output
3
5
传送门:【FZU】2171 防守阵地 II
题目大意:中文题,不说了。
题目分析:赤果果的区间更新,区间查询,秒之。
代码如下:
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std ;#define ls ( o << 1 )
#define rs ( o << 1 | 1 )
#define rt l , r , o
#define root 1 , n , 1
#define lson l , m , ls
#define rson m + 1 , r , rs
#define clear( A , X ) memset ( A , X , sizeof A )
#define mid ( ( l + r ) >> 1 )const int maxN = 100005 ;int sum[maxN << 2] , del[maxN << 2];
int L , R ;void PushUp ( int o ) {sum[o] = sum[ls] + sum[rs] ;
}void PushDown ( int l , int r , int o ) {if ( del[o] ) {int m = mid ;del[ls] += del[o] ;sum[ls] -= del[o] * ( m - l + 1 ) ;del[rs] += del[o] ;sum[rs] -= del[o] * ( r - m ) ;del[o] = 0 ;}
}void Build ( int l , int r , int o ) {del[o] = 0 ;if ( l == r ) {scanf ( "%d" , &sum[o] ) ;return ;}int m = mid ;Build ( lson ) ;Build ( rson ) ;PushUp ( o ) ;
}void Update ( int l , int r , int o ) {if ( L <= l && r <= R ) {++ del[o] ;sum[o] -= r - l + 1 ;return ;}PushDown ( rt ) ;int m = mid ;if ( L <= m ) Update ( lson ) ;if ( m < R ) Update ( rson ) ;PushUp ( o ) ;
}int Query ( int l , int r , int o ) {if ( L <= l && r <= R ) return sum[o] ;PushDown ( rt ) ;int m = mid , ans = 0 ;if ( L <= m ) ans += Query ( lson ) ;if ( m < R ) ans += Query ( rson ) ;return ans ;
}void work () {int n , m , q ;while ( ~scanf ( "%d%d%d" , &n , &m , &q ) ) {Build ( root ) ;while ( q -- ) {scanf ( "%d" , &L ) ;R = L + m - 1 ;printf ( "%d\n" , Query ( root ) ) ;Update ( root ) ;}}
}int main () {work () ;return 0 ;
}
这篇关于【FZU】2171 防守阵地 II 线段树的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!