【AtCoder-2164 C】Rabbit Exercise AtCoder【差分找规律+倍增】

题意:

一共 n n 个点,每个点都有自己的初始位置 x i x_i ,一共有 M M 次操作,每次操作给出一个 p o s pos ,则 a [ p o s ] = a [ p o s + 1 ] + a [ p o s 1 ] 2 a [ p o s ] a[pos]=a[pos+1]+a[pos-1]-2*a[pos] ,如果 p o s = 1 pos=1 a [ 1 ] = 2 a [ 2 ] a [ 1 ] a[1]=2*a[2]-a[1] ;如果 p o s = n pos=n a [ n ] = 2 a [ n 1 ] a [ n ] a[n]=2*a[n-1]-a[n] 。现将 M M 次操作称为一组操作,将该组操作执行 K K 次,求最后每个点的值。 ( 1 n , M 1 0 5 , 1 K 1 0 18 ) (1\leq n,M\leq 10^5,1\leq K\leq 10^{18})


思路:

该题的表象是数值变化,每操作一个点,就有一个点的数值发生变化。但我们要将一组操作执行 1 0 18 10^{18} 次,规模非常巨大,因此如果只是数值发生变化,我们难以求出如此大规模之后的答案,所以很明显一定要找到某种规律用更简单的方式来表达这种数值的变化。

一番挣扎过后…悄咪咪地看了题解…的确是想不出来,然后发现是通过差分来体现这种规律,具体规律如下。

f f 表示差分数组, f [ i ] = a [ i ] a [ i 1 ] f[i]=a[i]-a[i-1] 。操作一次 i i 位置,发生如下变化。
a [ i ] = a [ i 1 ] + a [ i + 1 ] a [ i ] f [ i ] = a [ i ] a [ i 1 ] = a [ i + 1 ] a [ i ] f [ i + 1 ] = a [ i + 1 ] a [ i ] = a [ i ] a [ i 1 ] a[i]'=a[i-1]+a[i+1]-a[i] \\ f[i]'=a[i]'-a[i-1]=a[i+1]-a[i] \\ f[i+1]'=a[i+1]-a[i]'=a[i]-a[i-1]
由上述三条式子可以发现,操作 1 1 个点的值,相当于在差分数组中将 i i 位置和 i + 1 i+1 位置的数进行了一次交换。由此我们得到了一次交换后的数值。

那如何得到 k k 次交换的数值呢?这里就涉及到了此题第二个巧妙的点,倍增!用倍增来快速求出大量重复性操作之后的答案。 p r e [ i ] [ j ] pre[i][j] 表示位置 i i 操作 2 j 2^j 组后对应的位置,则 p r e [ i ] [ j ] = p r e [ p r e [ i ] [ j 1 ] ] [ j 1 ] pre[i][j]=pre[pre[i][j-1]][j-1] ,由此即可进行倍增。


总结:

此题最重要的两个关键点。

  1. 通过差分数组将 “数值变化” 转换为 “交换变化”
  2. 通过倍增将大量重复性操作的复杂度变为 l o g log ,不由得感叹倍增的神奇

代码:

#include <bits/stdc++.h>
#define mem(a,b) memset(a,b,sizeof a);
#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 __ ios::sync_with_stdio(0);cin.tie(0);cout.tie(0)
typedef long long ll;
typedef double db;
const int N = 1e5+100;
const db EPS = 1e-9;
using namespace std;

void dbg() {cout << "\n";}
template<typename T, typename... A> void dbg(T a, A... x) {cout << a << ' '; dbg(x...);}
#define logs(x...) {cout << #x << " -> "; dbg(x);}

int n,m,a[N],pre[N][70];
ll x[N],k,v[N],ans[N];

void init(){
	for(ll j = 1; (1ll<<j) <= k; j++)
		for(int i = 0; i <= n; i++)
			pre[i][j] = pre[pre[i][j-1]][j-1];
}

int query(int pos,ll t){
	if(t == 0) return pos;
	for(ll j = 62; j >= 0; j--)
		if((1ll<<j) <= t) return query(pre[pos][j],t-(1ll<<j));
}

int main()
{
	scanf("%d",&n);
	rep(i,1,n) scanf("%lld",&x[i]);
	x[n+1] = x[n-1]; x[0] = x[2];
	v[0] = x[0];
	rep(i,1,n+1) v[i] = x[i]-x[i-1];
	rep(i,0,n+1) pre[i][0] = i;
	scanf("%d%lld",&m,&k);
	rep(i,1,m){
		scanf("%d",&a[i]);
		swap(pre[a[i]][0],pre[a[i]+1][0]);
	}
	init();
	rep(i,0,n+1) ans[i] = v[i];
	rep(i,1,n) ans[i] = v[query(i,k)];
	rep(i,1,n){
		ans[i] = ans[i-1]+ans[i];
		printf("%lld.0\n",ans[i]);
	}
	return 0;
}
发布了244 篇原创文章 · 获赞 115 · 访问量 3万+

猜你喜欢

转载自blog.csdn.net/qq_41552508/article/details/103150207