1005. Equal Sentences (dp / 字符串)2020 Multi-University Training Contest 4

传送门

1005. Equal Sentences (dp / 字符串)2020 Multi-University Training Contest 4_第1张图片
1005. Equal Sentences (dp / 字符串)2020 Multi-University Training Contest 4_第2张图片

思路:

  • 题意:给一个字符串 S ,求与其相似的本质不同的字符串 T 的个数。相似的定义有:

    • 单词的可重集合相同。
    • 对于一个词 α ,它在 S 中的第 i 次出现和在 T 中的第 i 次出现的下标相差不超过1。
  • 官方题解:
    1005. Equal Sentences (dp / 字符串)2020 Multi-University Training Contest 4_第3张图片

  • 根据题解,不难找到如下规律:

    • 如果S[i] == S[i-1],就有dp[i] = dp[i-1].
    • 否则就有dp[i] = dp[i-1] + dp[i-2] + 1.
    • 第 i 位置存在即是dp[i] + 1个解。

代码实现:

#include
#define endl '\n'
#define null NULL
#define ll long long
#define int long long
#define pii pair
#define lowbit(x) (x &(-x))
#define ls(x) x<<1
#define rs(x) (x<<1+1)
#define me(ar) memset(ar, 0, sizeof ar)
#define mem(ar,num) memset(ar, num, sizeof ar)
#define rp(i, n) for(int i = 0, i < n; i ++)
#define rep(i, a, n) for(int i = a; i <= n; i ++)
#define pre(i, n, a) for(int i = n; i >= a; i --)
#define IOS ios::sync_with_stdio(0); cin.tie(0);cout.tie(0);
const int way[4][2] = {{1, 0}, {-1, 0}, {0, 1}, {0, -1}};
using namespace std;
const int  inf = 0x7fffffff;
const double PI = acos(-1.0);
const double eps = 1e-6;
const ll   mod = 1e9 + 7;
const int  N = 2e5 + 5;

int t, n, dp[N];
string s[N];

signed main()
{
    IOS;

    cin >> t;
    while(t --){
        cin >> n;
        for(int i = 1; i <= n; i ++) cin >> s[i], dp[i] = 0;
        for(int i = 2; i <= n; i ++){ //第1位必须相同所以dp[1]=1,从2开始计算
            if(s[i] == s[i-1]) dp[i] = dp[i-1];
            else dp[i] = (dp[i-1] + dp[i-2] + 1) % mod;
        }
        cout << dp[n] + 1 << endl;
    }


    return 0;
}

你可能感兴趣的:(动态规划,比赛&训练,字符串)