6135.奶牛体检
题目链接:https://www.acwing.com/problem/content/description/6138/
题意:
给定两个长度为n的数组a,数组b。
操作一次:选定任意的l,r(其中1<=l<=r<=n),反转数组a的数字顺序
让你统计 在所有操作情况下 数组a与数组b 对应数字 数量 分别为0,1,···n的方案数
思路:
原本想从左至右暴力枚举,然而 每次 统计反转后对应数字的数量 时间复杂度O(n)这导致整体复杂度O(n*n*n),显然无法通过
通过回文想到 确定一个中点,然后 分别 向左/右两个方向 扩展区间,(遇到区间端点停止枚举),每次只要看 扩进去的区间 最左边 和 最右边 的值 是否 反转后与b数组对应。(因为之前的反转的结果不发生改变)
然而仅仅这样并没有枚举掉所有区间
发现加上 不枚举中点,即偶数情况的方案 答案才完整
实现方面利用前缀数组表示 原数组 与 b数组 对应的数字数量(ps:记得l可以等于r
#include<bits/stdc++.h> #define rep(i,a,n) for(int i=a;i<=n;i++) #define pb push_back #define endl "\n" #pragma GCC optimize(3) using namespace std; typedef long long ll; typedef pair<int,int> pii; const int inf=0x3f3f3f3f; const ll llmax=LLONG_MAX; const int maxn=1e5+5; const int mod=1e9+7; int n; int a[maxn]; int b[maxn]; int cnt[maxn]; int k[maxn]; signed main() { ios::sync_with_stdio(false),cin.tie(0); cin>>n; rep(i,1,n) cin>>a[i]; rep(i,1,n) cin>>b[i]; rep(i,1,n){ if(a[i]==b[i]){ k[i]++; } k[i]+=k[i-1]; } rep(i,2,n-1){ int l=i-1,r=i+1; int sum=0; if(a[i]==b[i])sum++; while(l>=1&&r<=n){ if(a[r]==b[l])sum++; if(a[l]==b[r])sum++; cnt[k[l-1]+(k[n]-k[r])+sum]++; l--;r++; } } rep(i,1,n-1){ int l=i,r=i+1; int sum=0; while(l>=1&&r<=n){ if(a[r]==b[l])sum++; if(a[l]==b[r])sum++; cnt[k[l-1]+(k[n]-k[r])+sum]++; l--;r++; } } rep(i,1,n){ cnt[k[n]]++; } rep(i,0,n){ cout<<cnt[i]<<endl; } return 0; }
【推荐】国内首个AI IDE,深度理解中文开发场景,立即下载体验Trae
【推荐】编程新体验,更懂你的AI,立即体验豆包MarsCode编程助手
【推荐】抖音旗下AI助手豆包,你的智能百科全书,全免费不限次数
【推荐】轻量又高性能的 SSH 工具 IShell:AI 加持,快人一步
· 分享一个免费、快速、无限量使用的满血 DeepSeek R1 模型,支持深度思考和联网搜索!
· 使用C#创建一个MCP客户端
· ollama系列1:轻松3步本地部署deepseek,普通电脑可用
· 基于 Docker 搭建 FRP 内网穿透开源项目(很简单哒)
· 按钮权限的设计及实现