121. 小红的区间翻转(卡码网周赛第二十五期(23年B站笔试真题))

题目链接
121. 小红的区间翻转(卡码网周赛第二十五期(23年B站笔试真题))

题目描述

小红拿到了两个长度为 n 的数组 a 和 b,她仅可以执行一次以下翻转操作:选择a数组中的一个区间[i, j],(i != j),将它们翻转。例如,对于 a = [2,3,4,1,5,6],小红可以选择左闭右闭区间[2,4],数组 a 则变成[2,3,5,1,4,6]。

小红希望操作后 a 数组和 b 数组完全相同。请你告诉小红有多少种操作的方案数。

初始 a 数组和 b 数组必定不相同。

输入

第一行输入一个正整数 n,代表数组的长度;

第二行输入 n 个正整数 ai;

第三行输入 n 个正整数 bi。

输出

选择区间的方案数。

样例1输入

4

1 2 3 1

1 3 2 1

样例1输出

2

样例2输入

4

1 1 1 1

1 1 1 1

样例2输出

6

样例3输入

10

19 2 13 71 14 14 71 40 16 23

19 2 13 71 14 14 71 40 16 23

样例3输出

2

提示

数据范围

1 ≤ n, ai ,bi ≤ 5000

在示例1中:

将 1 2 3 1 中的 2 3 进行翻转,得到 1 3 2 1。

将 1 2 3 1 整个进行翻转,得到 1 3 2 1。

所以最终结果是 2。


题解1(C++版本)

cpp 复制代码
// 区间dp
#include<bits/stdc++.h>
using namespace std;
const int N = 5010;

int n, a[N], b[N], ans;
bool dp1[N][N]; // dp1[i][j]为1表示将数组a的区间[i,j]进行翻转后,使得数组a和数组b这段区间相同
bool dp2[N][N]; // dp2[i][j]为1表示将数组a的区间[i,j]与数组b这段区间相同
int main(){
    scanf("%d", &n);
    
    for(int i = 1; i <= n; i++) scanf("%d", &a[i]);
    for(int i = 1; i <= n; i++) scanf("%d", &b[i]);
    for(int i = 1; i <= n; i++) dp2[i][i] = dp1[i][i] = true;
    for(int len = 2; len <= n; len++){ // 枚举区间长度
        for(int i = 1; i + len - 1 <= n; i++){ // 枚举区间左端点
            int j = i + len - 1; // 枚举区间右端点
            if((a[i] == b[j]) && (a[j] == b[i])){ // 将数组a的区间[i,j]进行翻转
                if(i + 1 < j) dp1[i][j] = dp1[i + 1][j - 1];
                else dp1[i][j] = true; // i和j相邻
            }
            if((a[i] == b[i]) && (a[j] == b[j])){ // 不将数组a的区间[i,j]进行翻转
                if(i + 1 < j) dp2[i][j] = dp2[i + 1][j - 1];
                else dp2[i][j] = true; // i和j相邻
            }
        }
    }
    for(int len = 2; len <= n; len++){
        for(int i = 1; i + len - 1 <= n; i++){
            int j = i + len - 1;
            bool f1 = true, f3 = true;
            if(i > 1) f1 = dp2[1][i - 1];
            if(j < n) f3 = dp2[j + 1][n];
            if(f1 && dp1[i][j] && f3) {
                ans++;
            }
        }
    }
    printf("%d\n", ans);
    return 0;
}

题解2(C++版本)

cpp 复制代码
// 字符串哈希
#include<bits/stdc++.h>
using namespace std;
typedef unsigned long long uLL;
 
const int N = 5005;
const uLL base = 2333;
 
uLL pw[N] = {1}, hs1[N], hs2[N], hs; // hs1表示正向求哈希,hs2表示反向求哈希, hs表示数组b的哈希
int n;
uLL a[N], b[N];
int main(){
    scanf("%d", &n);
    for(int i = 1; i <= n; i++){
        pw[i] = pw[i - 1] * base;
    }
    for(int i = 1; i <= n; i++) scanf("%u", &a[i]);
    for(int i = 1; i <= n; i++) scanf("%u", &b[i]), hs = hs * base + b[i];
    for(int i = 1; i <= n; i++) {
        hs1[i] = hs1[i - 1] * base + a[i];
    }
    for(int i = n ;i > 0; i--){
        hs2[i] = hs2[i + 1]*base + a[i];
    }
    //for(int i = 1; i <= n; i++) printf("%u ", hs1[i]);
    //printf("\n");
    // for(int i = 1; i <= n; i++) printf("%u ", hs2[i]);
    // printf("\n");
    int ans = 0;
    for(int i = 1; i < n; i++){
        for(int j = i + 1; j <= n; j++){
            uLL sum1 = hs1[i - 1]*pw[n - i + 1];
            uLL sum2 = (hs2[i] - hs2[j + 1]*pw[j - i + 1])*pw[n -j];
            uLL sum3 = (hs1[n] - hs1[j]*pw[n - j]);
            //printf("i = %d j = %d ", i, j);
            //printf("%u %u %u, %u \n", sum1, sum2, sum3, sum1 + sum2 + sum3);
            if(sum1 + sum2 + sum3 == hs) ans++;
        }
    }
    printf("%d\n", ans);
    return 0;
}
相关推荐
小码农<^_^>9 分钟前
优选算法精品课--滑动窗口算法(一)
算法
羊小猪~~11 分钟前
神经网络基础--什么是正向传播??什么是方向传播??
人工智能·pytorch·python·深度学习·神经网络·算法·机器学习
软工菜鸡37 分钟前
预训练语言模型BERT——PaddleNLP中的预训练模型
大数据·人工智能·深度学习·算法·语言模型·自然语言处理·bert
南宫生40 分钟前
贪心算法习题其三【力扣】【算法学习day.20】
java·数据结构·学习·算法·leetcode·贪心算法
AI视觉网奇1 小时前
sklearn 安装使用笔记
人工智能·算法·sklearn
JingHongB1 小时前
代码随想录算法训练营Day55 | 图论理论基础、深度优先搜索理论基础、卡玛网 98.所有可达路径、797. 所有可能的路径、广度优先搜索理论基础
算法·深度优先·图论
weixin_432702261 小时前
代码随想录算法训练营第五十五天|图论理论基础
数据结构·python·算法·深度优先·图论
小冉在学习2 小时前
day52 图论章节刷题Part04(110.字符串接龙、105.有向图的完全可达性、106.岛屿的周长 )
算法·深度优先·图论
Repeat7152 小时前
图论基础--孤岛系列
算法·深度优先·广度优先·图论基础
小冉在学习2 小时前
day53 图论章节刷题Part05(并查集理论基础、寻找存在的路径)
java·算法·图论