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;
}
posted @   Marinaco  阅读(13)  评论(0编辑  收藏  举报
相关博文:
阅读排行:
· 分享一个免费、快速、无限量使用的满血 DeepSeek R1 模型,支持深度思考和联网搜索!
· 使用C#创建一个MCP客户端
· ollama系列1:轻松3步本地部署deepseek,普通电脑可用
· 基于 Docker 搭建 FRP 内网穿透开源项目(很简单哒)
· 按钮权限的设计及实现
//雪花飘落效果
点击右上角即可分享
微信分享提示