nyoj 1205——简单问题——————【技巧题】

简单问题

时间限制： 1000 ms  |  内存限制：65535 KB

难度： 2

 

描述

给你一个n*m的矩阵，其中的元素每一行从左到右按递增顺序排序，每一列从上到下按递增顺序排序，然后给你一些数x，问这些书在不在这个矩阵中，若在，输出其列号最大的元素坐标。

 

输入

多组测试样例，首先两个数n，m（0<n,m<=1000）
然后n行m列的一个矩阵，其中的元素保证在32位整型范围
然后一个数cnt，表示询问数的个数（0<cnt<=100）
然后是cnt个要询问的元素

输出

如果存在，输出其列号最大的元素坐标
否则，输出“NO”

样例输入

4 4

1 2 8 9

2 4 9 12

4 7 10 13

6 8 11 15

3

7 8 14

样例输出

3 2

1 3

NO


解题思路：虽然用二分可以处理，但是用O（n+m）的复杂度过。由于矩阵是有特殊关系的。从每行左向右递增，每列从上向下递增。于是从第一行最右端起，如果元素大于要找的值，则往左移动，因为该列下边的值都比该元素，如果该元素小于要找的值，则向下移动，继续按照上边的方法判断，如果要找的位置不在矩阵内了，说明没有元素在矩阵中，否则输出位置即可。

 

#include<bits/stdc++.h>

using namespace std;

int a[2000][2000];

int main(){

    int n,m,q,aim,x,y;

    while(scanf("%d%d",&n,&m)!=EOF){

        for(int i=0;i<n;i++){

            for(int j=0;j<m;j++)

                scanf("%d",&a[i][j]);

        }

        scanf("%d",&q);

        for(int j=0;j<q;j++){

            scanf("%d",&aim);

            int y=m-1,x=0;

            bool flag=0;

            while(true){

                if(a[x][y]>aim){

                    y--;

                    if(y<0){

                        flag=1;

                        break;

                    }

                }else if(a[x][y]<aim){

                    x++;

                    if(x>=n){

                        flag=1;

                        break;

                    }

                }else{ break;}

            }

            if(flag){

                printf("NO\n");

            }else{

                printf("%d %d\n",x+1,y+1);

            }

        }

    }

    return 0;

}