蓝桥杯集训每日一题——Acwing 6135 奶牛体检

题目描述：

 代码1：不知道为什么一直超时，跟up主写的

#include<iostream>
#include<cstring>
#include<algorithm>

using namespace std;

const int N=7510;//多开位置防止数组越界 

int n;
int a[N],b[N],ans[N];
//a表示奶牛初始顺序，b表示兽医看的品种，ans表示匹配各种数量的方案数 

int main()
{	
	cin>>n;
	for(int i=1;i<=n;i++) cin>>a[i];
	for(int i=1;i<=n;i++) cin>>b[i];
	
	int cnt=0;//记录初始时奶牛匹配数 
	for(int i=1;i<=n;i++)
		if(a[i]==b[i])
			cnt++;
	
	//由于区间长度有奇数和偶数
	//奇数由1个点开始扩展，偶数由两个点开始扩展
	//即奇数是i和i+0向外扩展，偶数是i和i+1向外扩展		
	for(int i=1;i<=n;i++)
		for(int j=0;i<2;j++)
		{
			int sum=cnt;//存入初始值奶牛的匹配数 
			for(int l=i,r=i+j;l>=1&&r<=n;l--,r++)//从中心向两边扩展 
			{
				//减去交换之前对答案的贡献 
				if(a[l]==b[l]) sum--;
				if(a[r]==b[r]) sum--;
				//假的交换，即不交换，直接比较相反的位置 
				if(a[l]==b[r]) sum++;
				if(a[r]==b[l]) sum++;
				ans[sum]++;
			}
		}
	//输出每一种匹配的奶牛数量对应的方案数 
	for(int i=0;i<=n;i++)
		cout<<ans[i]<<endl;
	return 0;
 } 

代码2：

#include <iostream>

using namespace std;

const int N = 1e5 + 10;

int n;
int a[N], b[N], ans[N];

int main()
{
    cin >> n;
    for (int i = 1; i <= n; i ++) cin >> a[i];
    for (int i = 1; i <= n; i ++) cin >> b[i];

    int cnt = 0;
    for (int i = 1; i <= n; i ++) 
        if (a[i] == b[i]) cnt ++;

    for (int i = 1; i <= n; i ++) {
        int l = i, r = i;
        int sum = cnt;
        while (l >= 1 && r <= n) {
            sum += (a[l] == b[r]) + (a[r] == b[l]) - (a[l] == b[l]) - (a[r] == b[r]);
            ans[sum] ++;
            l --, r ++;
        }
        l = i - 1, r = i;
        sum = cnt;
        while (l >= 1 && r <= n) {
            sum += (a[l] == b[r]) + (a[r] == b[l]) - (a[l] == b[l]) - (a[r] == b[r]);
            ans[sum] ++;
            l --, r ++;
        }
    }

    for (int i = 0; i <= n; i ++) cout << ans[i] << '\n';

    return 0;
}

作者：acwing_98698
链接：https://www.acwing.com/solution/content/267539/
来源：AcWing
著作权归作者所有。商业转载请联系作者获得授权，非商业转载请注明出处。

方法2：

#include <iostream>

using namespace std;

const int N = 7500 + 10;

int n, cnt;
int a[N], b[N];
int dp[N][N], ans[N];

int main()
{
    cin >> n;
    for (int i = 1; i <= n; i ++) cin >> a[i];
    for (int i = 1; i <= n; i ++) cin >> b[i];

    for (int i = 1; i <= n; i ++) 
        if (a[i] == b[i]) cnt ++;

    for (int i = 1; i <= n; i ++) {
        dp[i][i] = cnt;
        dp[i + 1][i] = cnt;
        ans[cnt] ++;
    }


    for (int len = 2; len <= n; len ++) {
        for (int i = 1; i <= n - len + 1; i ++) {
            int j = i + len - 1;
            dp[i][j] = dp[i + 1][j - 1] + (a[i] == b[j]) + (a[j] == b[i]) - (a[i] == b[i]) - (a[j] == b[j]);
            ans[dp[i][i + len - 1]] ++;
        }
    }

    for (int i = 0; i <= n; i ++) 
        cout << ans[i] << '\n';

    return 0;
}

作者：acwing_98698
链接：https://www.acwing.com/solution/content/267539/
来源：AcWing
著作权归作者所有。商业转载请联系作者获得授权，非商业转载请注明出处。