ZCMU 1305: 第K完美序列

Description

 

有一个由n个数字组成的序列，序列的每个数字不是1就是2。如果一个序列是第K完美序列，则这个序列满足以下两点；
1.k是n的因子。
2.这个序列中每隔k个元素都相等。
例如序列(1,2,1,2,1,2,1,2)，不仅是一个第2完美序列而且又是一个第4完美序列。序列(1,1,1,1)是一个第1完美序列。
你的任务是给定一个由n个数字组成的序列，最少需要改变几个元素才能使这个序列变成一个第K完美序列。
Input
多组测试数据。
每组测试数据的第一行为两个正整数n,k(1<=k<=n<=100)。
第二行为n个由空格分隔的正整数代表这个序列。
Output
对于每组测试数据，输出最少需要改变的次数。
Sample Input
6 2
2 1 2 2 2 1
8 4
1 1 2 1 1 1 2 1
9 3
2 1 1 1 2 1 1 1 2
Sample Output
1
0
3

【分析】思路是把n个正整数序列拼接成n*(k/n)的矩阵，然后判断每一列需要改变的最小次数。如9 3 （2 1 1 1 2 1 1 1 2）为例，

           2  1   1

           1   2  1

           1   1  2

第一列数字2的个数为1，数字1的个数为2，所以需改变的数是2，改变次数为1，以此类推。

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <cmath>
using namespace std;
int main()
{
    int n,k;
    while(cin>>n>>k)
    {
        int x[n][n],a[n],b[n];
        memset(a,0,sizeof(a));
        memset(b,0,sizeof(b));
        int res=0;int i,j;
        for(i=0;i<n/k;i++)
        {
            for(j=0;j<k;j++)
            {
            cin>>x[i][j];
            }
        }


        for(j=0;j<k;j++)
        {
            for(i=0;i<n/k;i++)
            {
            if(x[i][j]==1)a[j]++;//1的个数
            else b[j]++;//2的个数
            }
        }

        for(int z=0;z<k;z++)
        {
            if(a[z]<b[z])res+=a[z];
            else res+=b[z];
        }
        cout<<res<<endl;
    }
    return 0;

}

之前提交了多次，都是时间超限，以为是二维数组的问题，后来换了个方法还是不行，最后抱着试试看的心态换成C++，把scanf和print替换成cin>> cout<<，两种方法都过了....神奇