首页 > 代码库 > HDU 5634 线段树
HDU 5634 线段树
题目链接:http://acm.hdu.edu.cn/showproblem.php?pid=5634
题意:给定一个长度为n的序列,有m次操作。操作有3种:
1 l,r :区间[l,r]的值变成phi[val[i]](l<=i<=r; phi是欧拉值)
2 l,r,x:区间[l,r]的值变成x
3 l,r:求区间[l,r]的和
思路:操作2和3就是传统的简单线段树,操作2对应区间覆盖,操作3对应区间求和,重点在于操作1,由于一个数经过不超过log次求phi后会变成1,所以可以在线段树是用一个same标记,如果整个区间的数都相同则操作1就转换成操作2的区间覆盖了。如果操作的区间[l,r]已经包含住当前递归的子树区间但是子树的same标记为假则继续递归到子树的same标记为真为止,最多递归到叶子结点。
#define _CRT_SECURE_NO_DEPRECATE #include<iostream> #include<cstdio> #include<cstring> #include<algorithm> #include<string> #include<queue> #include<vector> #include<time.h> #include<cmath> using namespace std; typedef long long int LL; #define L(k)(k<<1) #define R(k)(k<<1|1) const LL INF = 9223372036854775807; const int MAXN = 3e5 + 24; const int MAXX = 1e7 + 24; struct Node{ int l, r, val; LL sum; bool same; Node(int _l = 0, int _r = 0, int _val = 0, LL _sum = 0, bool _same = false){ l = _l; r = _r; sum = _sum; same = _same; val = _val; } }Seg[MAXN * 4]; int val[MAXN], Phi[MAXX]; void pushUp(int k){ Seg[k].sum = Seg[L(k)].sum + Seg[R(k)].sum; if (Seg[L(k)].val == Seg[R(k)].val&&Seg[L(k)].same&&Seg[R(k)].same&&Seg[L(k)].val != -1){ Seg[k].same = true; Seg[k].val = Seg[L(k)].val; } else{ Seg[k].same = false; Seg[k].val = -1; } } void pushDown(int k){ if (Seg[k].same){ Seg[L(k)].same = Seg[R(k)].same = true; Seg[L(k)].val = Seg[R(k)].val = Seg[k].val; Seg[L(k)].sum = 1LL * (Seg[L(k)].r - Seg[L(k)].l + 1)*Seg[L(k)].val; Seg[R(k)].sum = 1LL * (Seg[R(k)].r - Seg[R(k)].l + 1)*Seg[R(k)].val; } } void Build(int st, int ed, int k){ Seg[k].l = st; Seg[k].r = ed; Seg[k].same = false; Seg[k].val = -1; if (st == ed){ Seg[k].val = val[st]; Seg[k].sum = val[st]; Seg[k].same = true; return; } int mid = (st + ed) >> 1; Build(st, mid, L(k)); Build(mid + 1, ed, R(k)); pushUp(k); } void Change(int st, int ed, int val, int k){ if (Seg[k].l == st&&Seg[k].r == ed){ Seg[k].same = true; Seg[k].val = val; Seg[k].sum = 1LL * (ed - st + 1)*val; return; } pushDown(k); if (Seg[L(k)].r >= ed){ Change(st, ed, val, L(k)); } else if (Seg[R(k)].l <= st){ Change(st, ed, val, R(k)); } else{ Change(st, Seg[L(k)].r, val, L(k)); Change(Seg[R(k)].l, ed, val, R(k)); } pushUp(k); } void Modify(int st, int ed, int k){ if (Seg[k].l == st&&Seg[k].r == ed&&Seg[k].same){ Seg[k].val = Phi[Seg[k].val]; Seg[k].sum = 1LL * (ed - st + 1)*Seg[k].val; return; } pushDown(k); if (Seg[L(k)].r >= ed){ Modify(st, ed, L(k)); } else if (Seg[R(k)].l <= st){ Modify(st, ed, R(k)); } else{ Modify(st, Seg[L(k)].r, L(k)); Modify(Seg[R(k)].l, ed, R(k)); } pushUp(k); } LL Query(int st, int ed, int k){ if (Seg[k].l == st&&Seg[k].r == ed){ return Seg[k].sum; } pushDown(k); if (Seg[L(k)].r >= ed){ return Query(st, ed, L(k)); } else if (Seg[R(k)].l <= st){ return Query(st, ed, R(k)); } else{ return Query(st, Seg[L(k)].r, L(k)) + Query(Seg[R(k)].l, ed, R(k)); } pushUp(k); } void GetPhi(){ //预处理欧拉值 memset(Phi, 0, sizeof(Phi)); Phi[1] = 1; for (LL i = 2; i < MAXX; i++){ if (!Phi[i]){ for (LL j = i; j < MAXX; j += i){ if (!Phi[j]) Phi[j] = j; Phi[j] = Phi[j] / i*(i - 1); } } } } int main(){ //#ifdef kirito // freopen("in.txt", "r", stdin); // freopen("out.txt", "w", stdout); //#endif // int start = clock(); int n, t, m; GetPhi(); scanf("%d", &t); while (t--){ scanf("%d%d", &n, &m); for (int i = 1; i <= n; i++){ scanf("%d", &val[i]); } Build(1, n, 1); for (int i = 1; i <= m; i++){ int tpe, l, r, x; scanf("%d%d%d", &tpe, &l, &r); if (tpe == 1){ Modify(l, r, 1); } else if (tpe == 2){ scanf("%d", &x); Change(l, r, x, 1); } else{ printf("%lld\n", Query(l, r, 1)); } } } //#ifdef LOCAL_TIME // cout << "[Finished in " << clock() - start << " ms]" << endl; //#endif return 0; }
HDU 5634 线段树
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。