数据结构实验之数组二:稀疏矩阵
Problem Description
对于一个n*n的稀疏矩阵M(1 <= n <= 1000),采用三元组顺序表存储表示,查找从键盘输入的某个非零数据是否在稀疏矩阵中,如果存在则输出OK,不存在则输出ERROR。稀疏矩阵示例图如下:
0 14 0 0 -5
0 -7 0 0 0
36 0 0 28 0
Input
连续输入多组数据,每组数据的第一行是三个整数mu, nu, tu(tu<=50),分别表示稀疏矩阵的行数、列数和矩阵中非零元素的个数,数据之间用空格间隔,随后tu行输入稀疏矩阵的非零元素所在的行、列值和非零元素的值,每组数据的最后一行输入要查询的数据k。
Output
输出查询结果,查找成功输出OK,找不到输出ERROR。
Example Input
3 5 5
1 2 14
1 5 -5
2 2 -7
3 1 36
3 4 28
36
Example Output
OK
注
起初,我做这个题目时,用了二维数组,把题目中所给的非零元素都存在了数组中,然后再双重循环,一行一行查找,这样做也能输出结果,但是交上去就是超时。
#include<stdio.h>
#include<stdlib.h>
#include<string.h>
struct node
{
int a,b,c;//存储输入的tu行非零元素所在的行列与非零元素的值
}s[1111];
int main()
{
int i,j,k,p,q;
int mu,nu,tu;
int flag;
while(scanf("%d %d %d",&mu,&nu,&tu)!=EOF)
{
flag=0;// 标记是否查找成功,一定要在里面把flag 置0;
for(i=1;i<=tu;i++)
scanf("%d %d %d",&s[i].a,&s[i].b,&s[i].c);
scanf("%d",&k);
for(i=1;i<=tu;i++)
{
if(s[i].c==k)//查找元素
{
flag=1;
printf("OK\n");;
break;
}
}
if(flag==0)
printf("ERROR\n");;
}
return 0;
}
超时代码
#include<stdio.h>
#include<stdlib.h>
#include<string.h>
int vis[1001][1001];
int main()
{
int mu,nu,tu,i,j,k,s,p,q;
int flag=0;
memset(vis,0,sizeof(vis));
scanf("%d %d %d",&mu,&nu,&tu);
for(k=1;k<=tu;k++)
{
scanf("%d %d %d",&i,&j,&s);
vis[i][j]=s;
}
scanf("%d",&p);
for(k=1;k<=mu;k++)
{
for(q=1;q<=nu;q++)
{
if(vis[k][q]==p)
flag=1;
}
}
if(flag==1)
printf("OK\n");
else
printf("ERROR\n");
}