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;
}
相关推荐
The_Ticker12 分钟前
CFD平台如何接入实时行情源
java·大数据·数据库·人工智能·算法·区块链·软件工程
爪哇学长1 小时前
双指针算法详解:原理、应用场景及代码示例
java·数据结构·算法
Dola_Pan1 小时前
C语言:数组转换指针的时机
c语言·开发语言·算法
繁依Fanyi1 小时前
简易安卓句分器实现
java·服务器·开发语言·算法·eclipse
烦躁的大鼻嘎1 小时前
模拟算法实例讲解:从理论到实践的编程之旅
数据结构·c++·算法·leetcode
C++忠实粉丝2 小时前
计算机网络socket编程(4)_TCP socket API 详解
网络·数据结构·c++·网络协议·tcp/ip·计算机网络·算法
用户37791362947552 小时前
【循环神经网络】只会Python,也能让AI写出周杰伦风格的歌词
人工智能·算法
福大大架构师每日一题2 小时前
文心一言 VS 讯飞星火 VS chatgpt (396)-- 算法导论25.2 1题
算法·文心一言
EterNity_TiMe_2 小时前
【论文复现】(CLIP)文本也能和图像配对
python·学习·算法·性能优化·数据分析·clip
机器学习之心3 小时前
一区北方苍鹰算法优化+创新改进Transformer!NGO-Transformer-LSTM多变量回归预测
算法·lstm·transformer·北方苍鹰算法优化·多变量回归预测·ngo-transformer