Codeforces 455D Serega and Fun(分块)

题意:

给出一个长度为 n n n 的序列 a 1 , a 2 , . . . , a n a_1,a_2,...,a_n a1,a2,...,an,q此操作:

  • 将区间 [ l , r ] [l,r] [l,r] 循环移位 ( a l → a l + 1 , a l + 1 → a l + 2 , . . . , a r → a l ) (a_l\rightarrow a_{l+1},a_{l+1}\rightarrow a_{l+2},...,a_r\rightarrow a_l) (alal+1,al+1al+2,...,aral)

  • 询问区间 [ l , r ] [l,r] [l,r] 内等于 k k k 的数有多少个

强制在线

这个题目比较新颖

循环移位怎么维护? 分块确实也可以

我们对每个块维护一个双端队列,对于每一次循环移位

  • 整块只需要一边进一边出就行了
  • 不完整暴力移位

如何统计答案?

显然,又是分块题的老套路

f [ i ] [ j ] f[i][j] f[i][j] 表示前 i i i 个块数 j j j 出现的次数,可以预处理

维护因为每个整块只修改两个数,所以依然可以维护 f [ i ] [ j ] f[i][j] f[i][j]

对于不完整块,直接暴力统计即可

时间复杂度 O ( q n ) O(q\sqrt n) O(qn )

#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
using namespace std ;
//#define int long long
#define rep(i, a, b) for (int i = (a); i <= (b); i++)
#define per(i, a, b) for (int i = (a); i >= (b); i--)
#define loop(s, v, it) for (s::iterator it = v.begin(); it != v.end(); it++)
#define cont(i, x) for (int i = head[x]; i; i = e[i].nxt)
#define clr(a) memset(a, 0, sizeof(a))
#define ass(a, sum) memset(a, sum, sizeof(a))
#define lowbit(x) (x & -x)
#define all(x) x.begin(), x.end()
#define ub upper_bound
#define lb lower_bound
#define pq priority_queue
#define mp make_pair
#define pb push_back
#define fi first
#define se second
#define iv inline void
#define enter cout << endl
#define siz(x) ((int)x.size())
#define file(s) freopen(s".in", "r", stdin), freopen(s."out", "w", stdout)
typedef long long ll ;
typedef unsigned long long ull ;
typedef pair <int, int> pii ;
typedef vector <int> vi ;
typedef vector <pii> vii ;
typedef queue <int> qi ;
typedef queue <pii> qii ;
typedef deque <int> di ;
typedef set <int> si ;
typedef map <int, int> mii ;
typedef map <string, int> msi ;
const int N = 100010 ;
const int M = 350 ;
const int INF = 0x3f3f3f3f ;
const int iinf = 1 << 30 ;
const ll linf = 2e18 ;
const int MOD = 1000000007 ;
const double eps = 1e-7 ;
void print(int x) { cout << x << endl ; exit(0) ; }
void PRINT(string x) { cout << x << endl ; exit(0) ; }
void douout(double x){ printf("%lf\n", x + 0.0000000001) ; }

int n, m, len, num, ans = 0 ;
int b[M][N] ;
di d[M] ;
int a[N], bl[N], l[M], r[M] ;

void build() {
	len = sqrt(n), num = (n - 1) / len + 1 ;
	rep(i, 1, n) bl[i] = (i - 1) / len + 1 ;
	rep(i, 1, num) l[i] = (i - 1) * len + 1, r[i] = i * len ;
	rep(i, 1, n) {
		b[bl[i]][a[i]]++ ;
		d[bl[i]].pb(a[i]) ;
	}
}

void modify(int x, int y) {
	if (bl[x] == bl[y]) {
		int z = bl[x] ;
		x -= l[z], y -= l[z] ;
		int tmp = d[z][y] ;
		per(i, y, x + 1) d[z][i] = d[z][i - 1] ;
		d[z][x] = tmp ;
	} else {
		int tmp = d[bl[y]][y - l[bl[y]]],
		tmp2 = d[bl[x]][r[bl[x]] - l[bl[x]]] ;
		b[bl[x]][tmp2]-- ; b[bl[y]][tmp]-- ;
		per(i, r[bl[x]], x + 1) {
			d[bl[x]][i - l[bl[x]]] = d[bl[x]][i - l[bl[x]] - 1] ;
		}
		d[bl[x]][x - l[bl[x]]] = tmp ;
		b[bl[x]][tmp]++ ;
		rep(i, bl[x] + 1, bl[y] - 1) {
			int tp = tmp2 ;
			b[i][tmp2]++ ;
			tmp2 = d[i][r[i] - l[i]] ;
			d[i].pop_back() ;
			d[i].push_front(tp) ;
			b[i][tmp2]-- ;
		}
		per(i, y, l[bl[y]] + 1) {
			d[bl[y]][i - l[bl[y]]] = d[bl[y]][i - l[bl[y]] - 1] ;
		}
		d[bl[y]][0] = tmp2 ;
		b[bl[y]][tmp2]++ ;
	}
}

int query(int x, int y, int val) {
	int ans = 0 ;
	if (bl[x] == bl[y]) {
		rep(i, x, y) if (d[bl[x]][i - l[bl[x]]] == val) ans++ ;
	} else {
		rep(i, x, r[bl[x]]) {
			if (d[bl[x]][i - l[bl[x]]] == val) ans++ ;
		}
		rep(i, bl[x] + 1, bl[y] - 1) ans += b[i][val] ;
		rep(i, l[bl[y]], y) {
			if (d[bl[y]][i - l[bl[y]]] == val) ans++ ;
		}
	}
	return ans ;
}

signed main(){
	scanf("%d", &n) ;
    rep(i, 1, n) scanf("%d", &a[i]) ;
    build() ;
    scanf("%d", &m) ;
	while (m--) {
		int op, x, y ; scanf("%d%d%d", &op, &x, &y) ;
		x = (x + ans - 1) % n + 1, y = (y + ans - 1) % n + 1 ;
		if (x > y) swap(x, y) ;
		if (op == 1) modify(x, y) ;
		else {
			int val ; scanf("%d", &val) ;
			val = (val + ans - 1) % n + 1 ;
			ans = query(x, y, val) ;
			printf("%d\n", ans) ;
		}
	}
	return 0 ;
}

/*
写代码时请注意:
	1.ll?数组大小,边界?数据范围?
	2.精度?
	3.特判?
	4.至少做一些
思考提醒:
	1.最大值最小->二分?
	2.可以贪心么?不行dp可以么
	3.可以优化么
	4.维护区间用什么数据结构?
	5.统计方案是用dp?模了么?
	6.逆向思维?
*/


你可能感兴趣的:(分块,———数据结构———)