【setDS】牛客小白月赛83 E

登录---专业IT笔试面试备考平台_牛客网

题意

思路

首先,一个必要步骤是把它转化为两个序列,这样就变成了一个序列DS问题

我们的答案是一个位置 pos 后面还有多少位置和这个位置的颜色相同,考虑得到这个答案我们需要维护什么东西

我们只需要维护颜色之间的边界的位置即可

用 set 维护位置,剩下就是常规的增删改查了

cpp 复制代码
#include <bits/stdc++.h>

#define int long long

constexpr int N = 5e2 + 10;
constexpr int M = 1e4 + 10;
constexpr int mod = 998244353;
constexpr int Inf = 0x3f3f3f3f;

std::set<int> sa, sb;

int n, m;

int a(int i, int j) {
	if (i & 1) {
		return (i - 1) * m + j;
	}else {
		return i * m - (j - 1);
	}
}
int b(int i, int j) {
	if (j & 1) {
		return (j - 1) * n + i;
	}else {
		return j * n - (i - 1);
	}
}
void upd(std::set<int> &S, std::vector<char> &V, int i, char x) {
	S.erase(i);
	S.erase(i + 1);
	V[i] = x;
	if (V[i] != V[i - 1]) {
		S.insert(i);
	}
	if (V[i] != V[i + 1]) {
		S.insert(i + 1);
	}
}
void solve() {
	std::cin >> n >> m;
	std::vector<char> va(n * m + 2), vb(n * m + 2);
	for (int i = 1; i <= n; i ++) {
		for (int j = 1; j <= m; j ++) {
			char x;
			std::cin >> x;
			upd(sa, va, a(i, j), x);
			upd(sb, vb, b(i, j), x);
		}
	}
	int q;
	std::cin >> q;
	while(q --) {
		int op, x, y;
		char c;
		std::cin >> op;
		if (op == 1) {
			std::cin >> x >> y >> c;
			upd(sa, va, a(x, y), c);
			upd(sb, vb, b(x, y), c);
		}else if (op == 2) {
			std::cin >> x >> y;
			std::cout << (*sa.upper_bound(a(x, y))) - a(x, y) << "\n";
		}else {
			std::cin >> x >> y;
			std::cout << (*sb.upper_bound(b(x, y))) - b(x, y) << "\n";
		}
	}
}
signed main(){
    std::ios::sync_with_stdio(false);
    std::cin.tie(nullptr);

    int t = 1;
    while(t --) {
        solve();
    }
    return 0;
}
相关推荐
大数据魔法师13 分钟前
基于Pandas和FineBI的昆明职位数据分析与可视化实现(五) - 基于随机森林算法预测职位分类
算法·pandas
楼田莉子15 分钟前
数据学习之队列
c语言·开发语言·数据结构·学习·算法
让我们一起加油好吗43 分钟前
【基础算法】贪心 (四) :区间问题
c++·算法·贪心算法·洛谷
孤独的追光者1 小时前
论文阅读|汽车虚拟环绕音响系统设计与实现策略的比较研究
算法·汽车·音频·信号处理·数字信号处理
黄皮の电气鼠2 小时前
C++:继承
开发语言·c++·算法
shylyly_2 小时前
专题一_双指针_查找总价格为目标值的两个商品
c++·算法·leetcode·双指针·查找总价格为目标值的两个商品·和为s的两个数
zzj_2626102 小时前
高精度乘法
算法
我尽力学2 小时前
HashMap的get、put流程源码分析
算法·哈希算法
IT猿手2 小时前
2025最新智能优化算法:沙狐优化(Rüppell‘s Fox Optimizer,RFO)算法求解23个经典函数测试集,完整MATLAB代码
android·算法·matlab·迁移学习·优化算法·动态多目标优化·动态多目标进化算法
嗜好ya3 小时前
LeetCode 560: 和为K的子数组
数据结构·算法·leetcode