AGC006 C Rabbit Exercise——思路（置换）

题目：https://agc006.contest.atcoder.jp/tasks/agc006_c

选了 i 位置后 x[ i ] = x[ i-1 ] + x[ i+1 ] - x[ i ] 。

从差分序列来看，选 i 位置后就是 swap( a[ i ] , a[ i+1 ] ) 。

一组之后就是一个置换。找一下每个位置所在的环就行了。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#define ll long long
using namespace std;
const int N=1e5+;
int n,a[N],b[N],c[N],tot,sta[N<<],lcm; ll k;
bool vis[N];
int rdn()
{
  int ret=;bool fx=;char ch=getchar();
  while(ch>''||ch<''){if(ch=='-')fx=;ch=getchar();}
  while(ch>=''&&ch<='') ret=ret*+ch-'',ch=getchar();
  return fx?ret:-ret;
}
int gcd(int a,int b){return b?gcd(b,a%b):a;}
void dfs(int cr)
{
  sta[++tot]=cr;vis[cr]=;
  if(!vis[b[cr]])dfs(b[cr]);
}
int main()
{
  n=rdn();
  for(int i=;i<=n;i++)a[i]=rdn();
  for(int i=n;i;i--)a[i]-=a[i-];
  for(int i=;i<=n;i++)b[i]=i;
  int m;m=rdn(); scanf("%lld",&k);
  for(int i=,d;i<=m;i++)
    {
      d=rdn(); swap(b[d],b[d+]);
    }
  lcm=;
  for(int i=;i<=n;i++)
    if(!vis[i])
      {
    tot=; dfs(i);
    for(int j=tot+;j<=tot<<;j++)
      sta[j]=sta[j-tot];
    int d=k%tot;
    for(int j=;j<=tot;j++)
      c[sta[j]]=a[sta[j+d]];
      }
  ll lj=c[];
  for(int i=;i<=n;i++,lj+=c[i])
    printf("%lld\n",lj);
  return ;
}