首页 > 代码库 > HLG 1808 绘画 (哈希 + map + 线段树)
HLG 1808 绘画 (哈希 + map + 线段树)
链接: http://acm.hrbust.edu.cn/index.php?m=ProblemSet&a=showProblem&problem_id=1808
Description
小胖子最近喜欢上了画画,但是他的画画技术很差,只能从零开始,小胖子在一条线上画了很多彩色的球,但是觉得不好看,就想修改。
小胖子站在画前想了又想,他有时会将一些连续的点涂成相同的颜色。但他感觉累的时候,就会无聊的数数某个颜色在某段区间内出现的次数。
Input
每组数据的第一行输入两个正整数n和m(1<-n,m<=100000),n代表彩球的个数长度,m代表询问次数。
接下来一行包括n个整数代表每个位置的颜色。
接下来m行,每行包括四个正整数a,l,r,z(1<=a<=2, 0<=l<=r<n ,0<=z<2^31)
a=1代表将l到r之间的彩球涂成z;
a=2代表小胖子想知道有多少个颜色为z的彩球在l到r出现了多少次。
Output
输出每次询问的答案。
Sample Input
5 5
1 2 3 4 0
2 1 3 3
1 1 3 1
2 1 3 3
2 0 3 1
2 3 4 1
Sample Output
1
0
4
1
代码及解析如下:
#include <iostream> #include <cstdio> #include <cmath> #include <cstring> #include <cstdlib> #include <map> #include <algorithm> #define MAXN 100005 #define RST(N)memset(N, 0, sizeof(N)) using namespace std; int n, m ,bsize, bnum, x[MAXN]; int q, l, r, z; struct hash_block { int cls, size; std::map <int, int> mp; }b[350]; //下传标记,当这个块化整为零的时候需要下传标记并跟新所有元素 void pushdown(int id) { hash_block &hb = b[id]; if(hb.cls != -1) { for(int i=id*bsize; i<id*bsize+hb.size; i++) x[i] = hb.cls; hb.mp.clear(), hb.mp[hb.cls] = hb.size; hb.cls = -1; } } //更新,中间的部分打标记就可以了,两边的sqrt(n)暴力更新 void update(int l, int r, int c) { int lb = l/bsize, rb = r/bsize, ans = 0; for(int i=lb+1; i<rb; i++) b[i].cls = c; if(lb != rb) { pushdown(lb), pushdown(rb); for(int i=l; i<lb*bsize+b[lb].size; i++) { b[lb].mp[x[i]]--; b[lb].mp[c]++; x[i] = c; } for(int i=rb*bsize; i<=r; i++) { b[rb].mp[x[i]]--; b[rb].mp[c]++; x[i] = c; } }else { pushdown(lb); for(int i=l; i<=r; i++) { b[lb].mp[x[i]]--; b[lb].mp[c]++; x[i] = c; } } return ; } //中间的部分根据标记或者hash表可以直接查询,两边的sqrt(n)暴力查询 int query(int l, int r, int c) { int lb = l/bsize, rb = r/bsize, ans = 0; for(int i=lb+1; i<rb; i++) { //一直错在这个地方了,如果有标记,直接判断标记是不是需要的颜色 //如果没标记,要先判map中有没有这个元素然后在操作,否则会MLE! if(b[i].cls == c) ans += b[i].size; else if(b[i].cls == -1 && b[i].mp.find(c) != b[i].mp.end()) ans += b[i].mp[c]; } if(lb != rb) { pushdown(lb), pushdown(rb); for(int i=l; i<lb*bsize+b[lb].size; i++) ans += (x[i]==c); for(int i=rb*bsize; i<=r; i++) ans +=( x[i]==c); }else { pushdown(lb); for(int i=l; i<=r; i++) ans += (x[i]==c); } return ans; } void initblock() { bsize = (int)sqrt(n+1e-8); bnum = (n-1)/bsize + 1; for(int i=0; i<bnum; i++) { b[i].mp.clear(); b[i].cls = -1; b[i].size = std::min(i*bsize+bsize, n) - i*bsize; } for(int i=0; i<n; i++){ scanf("%d", &x[i]); b[i/bsize].mp[x[i]]++; } } int main() { while(~scanf("%d %d", &n, &m)) { initblock(); while(m--) { scanf("%d %d %d %d", &q, &l, &r, &z); if(q == 1) update(l, r, z); else printf("%d\n", query(l, r, z)); } } return 0; }
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。