hnu_12450 Painting(杂7杂8)

http://acm.hnu.cn/online/?action=problem&type=show&id=12450&courseid=229

题意：

给你一个n*n的矩阵代表一面墙，有k种颜色去刷这面墙，可以横着刷可以坚着刷，但要保证用k颜色去刷
一行或一列，这一行或一列的至少要有两个与k颜色。每刷一次记一次操作，求最小操作数。若用多种颜色刷都可以
达到最小，从小到大输出颜色。 
思路：
没什么算法啊，纯暴力啊，直接模拟啊，要分很多种情况啊，各种WA啊。。。

首先确定刷的次数只可能是n,n+1,n+2或者无解，再分情况讨论即可。（PS：代码写得好臃肿） 

我的代码：

/*
program:hnu_12450 
author:BlackAndWhite
*/ 
#include<stdio.h>
#include<algorithm>
#include<string.h>
int a[505][505],n,k,T;
int i,j,t,q;
struct node
{
    int id,val;
}ans[505];
int fun(int k,int a[505][505])
{
    int i,j,v[505],h[505],V=0,H=0,t1,t2;
    memset(v,0,sizeof(v));
    memset(h,0,sizeof(h));
    for(i=0;i<n;i++)
    {
        for(j=0;j<n;j++)
            if(k==a[i][j])
            {
                v[i]++;
                if(v[i]>=2){V++;break;}
            }
    }
    for(j=0;j<n;j++)
    {
        for(i=0;i<n;i++)
            if(k==a[i][j])
            {
                h[j]++;
                if(h[j]>=2){H++;break;}
            }
    }
    if(H==n||V==n) return n;
    if(!H&&!V) return 0;
    if(H>1||V>1)//有解的情况 
    {
        t1=t2=0x7fffffff;//初始化啊啊啊，WA了N次 
        if(H>1)
        {
        for(i=0;i<n;i++)
            if(!h[i]) {t1=n+2;break;}
        if(i==n) t1=n+1;
        }
        if(V>1)
        {
        for(i=0;i<n;i++)
            if(!v[i]) {t2=n+2;break;}
        if(i==n) t2=n+1;
        }
        if(t1<t2) return t1;
        return t2;
    }
    if(H==1&&V==0)// 不一定有解的情况 
    {
        int z1=0,z2=0;
        for(i=0;i<n;i++)
            if(h[i]) z1++;
        for(i=0;i<n;i++)
            if(v[i]) z2++;
        if(z1>=3&&z2>=4) return n+2;
        return 0;
    }
    if(H==0&&V==1)// 不一定有解的情况
    {
        int z1=0,z2=0;
        for(i=0;i<n;i++)
            if(h[i]) z1++;
        for(i=0;i<n;i++)
            if(v[i]) z2++;
        if(z1>=4&&z2>=3) return n+2;
        return 0;
    }
    if(H==1&&V==1)// 不一定有解的情况
    {
        int z1,z2;
        for(z1=0;z1<n;z1++) if(v[z1]>1) break;
        for(z2=0;z2<n;z2++) if(h[z2]>1) break;
        for(i=0;i<n;i++)
            for(j=0;j<n;j++)
                if(a[i][j]==k&&i!=z1&&j!=z2) return n+2;
        return 0;
    }
    return 0;
}
int cmp(node a,node b)
{
    if(a.val==b.val) return a.id<b.id;
    return a.val<b.val;
}
int main()
{
    scanf("%d",&T);
    while(T--)
    {
        scanf("%d%d",&n,&k);
        for(i=0;i<n;i++)
            for(j=0;j<n;j++)
                scanf("%d",&a[i][j]);
        for(i=0;i<k;i++) ans[i].id=i+1;
        for(i=0;i<k;i++)
            ans[i].val=fun(i+1,a);
        std::sort(ans,ans+k,cmp);
        for(i=0;i<k;i++) if(ans[i].val) break;
        if(i==k) {printf("0\n");continue;}
        printf("%d\n",ans[i].val);
        ans[k].val=-1;
        for(j=i;j<k;j++) if(ans[j].val!=ans[j+1].val) break;
        j++;
        for(q=i;q<j-1;q++)
            printf("%d ",ans[q].id);
        printf("%d\n",ans[q].id);
    }
    return 0;
}
/*
10
8 10
1 8 8 6 6 5 8 10 
3 3 1 1 9 5 4 3 
1 1 4 7 9 6 3 9 
2 6 9 9 2 10 1 2 
9 1 7 1 10 6 6 5 
1 9 8 10 5 3 9 8 
1 6 8 3 10 1 5 6 
3 8 7 7 7 7 4 3 

5 10
2 2 1 8 4 
9 8 2 1 9 
1 5 8 3 5 
3 10 7 5 5 
10 8 1 4 6 

7 10
7 3 7 4 10 9 3 
2 6 5 6 10 2 2 
3 9 4 10 2 1 7 
3 5 10 3 5 9 2 
9 6 7 9 2 6 1 
2 9 6 5 1 8 10 
2 9 8 3 10 5 7 
*/