1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69
| struct SegTree {
struct Node { int l, r; ll sum; ll mx; } tr[maxn << 2];
void push_up( int u ) { tr[u].sum = tr[ls].sum + tr[rs].sum; tr[u].mx = max( tr[ls].mx, tr[rs].mx ); }
void build( int u, int l, int r, ll val[] ) { tr[u] = { l, r, 0, 0 }; if( l == r ) { tr[u].sum = val[l]; tr[u].mx = val[l]; return ; } int mid = ( l + r ) >> 1; build( ls, l, mid, val ); build( rs, mid + 1, r, val ); push_up( u ); }
void upd( int u, int l, int r ) { if( tr[u].mx <= 1 ) return ; if( tr[u].l == tr[u].r ) { tr[u].sum = sqrt( tr[u].sum ); tr[u].mx = tr[u].sum; return ; } int mid = ( tr[u].l + tr[u].r ) >> 1; if( l <= mid ) upd( ls, l, r ); if( mid < r ) upd( rs, l, r ); push_up( u ); }
ll ask( int u, int l, int r ) { if( l <= tr[u].l && tr[u].r <= r ) return tr[u].sum; int mid = ( tr[u].l + tr[u].r ) >> 1; ll res = 0; if( l <= mid ) res += ask( ls, l, r ); if( mid < r ) res += ask( rs, l, r ); return res; }
} seg;
ll a[maxn];
void solve( ) { int n = read( ); for( int i = 1; i <= n; ++ i ) a[i] = read( ); seg.build( 1, 1, n, a );
int m = read( ); for( int i = 1; i <= m; ++ i ) { int opt = read( ); int l = read( ), r = read( ); if( l > r ) swap( l , r ); if( opt == 0 ) { seg.upd( 1, l, r ); } else { cout << seg.ask( 1, l, r ) << '\n'; } } }
|