洛谷题解：AT_agc031_b Reversi

题目传送门（洛谷版）

个人感觉本题还是比较水的，就是一道新手动态规划，顶多黄。

本文定义

$f_i$：表示到第 $i$ 块石头有几种方案。
$Las{t}_i$：表示上一块颜色 $i$ 的位置。
注：后文所有 $Las{t}_i$ 均用 $l_i$ 代替。
$a_i$：表示第 $i$ 块石头颜色。

思路

对于每块石头，我们都有两种情况，就是染色与不染色。

• 第一种：不染色。直接让 $f_i$ 继承上一块石头的值。

• 第二种：染色。$f_i$ 就是 $f_i-1+f_{l}i$。

值得注意的地方是每次计算后就要莫一下。

$l_i$ 的值每次更新为 $a_i$。

#include<bits/stdc++.h>
using namespace std;
int n,i,j,a[200002],last[200002],f[200002];
int main()
{
  ios::sync_with_stdio(false);
  cin.tie(0);
  cout.tie(0);
  cin>>n;
  for(i=1;i<=n;++i)
  cin>>a[i];
  f[1]=1;//单独一块石头就只有一种情况，无论染或不染
  last[a[1]]=1;//第一块石头第一次出现在1
  for(i=2;i<=n;++i)//从第二块石头开始dp
  {
  	f[i]=f[i-1]%1000000007;//第一种情况
  	if(a[i]!=a[i-1]&&last[a[i]])//第二种
  	f[i]=(f[i]+f[last[a[i]]])%1000000007;
//  	else if(a[i]==a[i-1])last[a[i]]=i;
  	last[a[i]]=i;
  }
  cout<<f[n]%1000000007;
  return 0;
}