优化这段代码 int main() { SqStack s; int choose, flag = 0; SElemType j, e, t; cout << "1.初始化\n"; cout << "2.入栈\n"; cout << "3.读栈顶元素\n"; cout << "4.出栈\n"; cout << "0.退出\n\n"; choose = -1; while (choose != 0) { cout << "请选择:"; cin >> choose; switch (choose) { case 1: if (InitStack(s)) { flag = 1; cout << "成功对栈进行初始化\n\n"; } else cout << "初始化栈失败\n\n"; break; case 2: { fstream file; file.open("SqStack.txt"); if (!file) { cout << "错误!未找到文件!\n\n" << endl; exit(ERROR); } if (flag) { flag = 1; cout << "进栈元素依次为:\n"; while (!file.eof()) { file >> j; if (file.fail()) break; else { Push(s, j); cout << j << " "; } } cout << endl << endl; } else cout << "栈未建立,请重新选择\n\n"; file.close(); } break; case 3://算法3.3 顺序栈的出栈 if(flag != -1 && flag != 0) cout << "栈顶元素为:\n" << GetTop(s) << endl << endl; else cout << "栈中无元素,请重新选择\n" << endl; break; case 4://算法3.4 取顺序栈的栈顶元素 cout << "依次弹出的栈顶元素为:\n"; while (Pop(s, t)){ flag = -1; cout << t << " "; } cout << endl << endl; break; } } return 0; }
时间: 2024-01-14 09:05:06 浏览: 77
这段代码的问题主要在于:
1. 在第2个case中,每次打开文件都需要进行文件查找,这样会影响程序的效率。
2. 在第3个case中,没有对栈为空的情况进行判断,如果栈为空,会出现错误。
3. 在第4个case中,当栈为空时,flag的值会被修改为-1,但是在第3个case中,却使用了flag来判断栈是否为空,这样会产生混淆。
以下是优化后的代码:
```c++
int main()
{
SqStack s;
int choose, flag = 0;
SElemType j, e, t;
cout << "1.初始化\n";
cout << "2.入栈\n";
cout << "3.读栈顶元素\n";
cout << "4.出栈\n";
cout << "0.退出\n\n";
choose = -1;
while (choose != 0)
{
cout << "请选择:";
cin >> choose;
switch (choose)
{
case 1:
if (InitStack(s))
{
flag = 1;
cout << "成功对栈进行初始化\n\n";
}
else
cout << "初始化栈失败\n\n";
break;
case 2:
{
fstream file;
file.open("SqStack.txt", ios::in);
if (!file)
{
cout << "错误!未找到文件!\n\n" << endl;
exit(ERROR);
}
if (flag)
{
flag = 1;
cout << "进栈元素依次为:\n";
while (file >> j)
{
Push(s, j);
cout << j << " ";
}
cout << endl << endl;
}
else
cout << "栈未建立,请重新选择\n\n";
file.close();
}
break;
case 3:// 算法3.3 顺序栈的读取
if (!StackEmpty(s))
cout << "栈顶元素为:\n" << GetTop(s) << endl << endl;
else
cout << "栈中无元素,请重新选择\n" << endl;
break;
case 4:// 算法3.4 顺序栈的出栈
if (!StackEmpty(s))
{
cout << "依次弹出的栈顶元素为:\n";
while (Pop(s, t))
{
flag = 1;
cout << t << " ";
}
cout << endl << endl;
}
else
cout << "栈中无元素,请重新选择\n" << endl;
break;
}
}
return 0;
}
```
优化后的代码主要做了以下改进:
1. 在第2个case中,只有在flag为1的情况下,才需要进行文件读取操作,这样可以减少文件查找的次数。
2. 在第3个case和第4个case中,都对栈是否为空进行了判断,避免了出现错误。
3. 在第4个case中,当栈不为空时,将flag的值修改为1,这样就不会与第3个case产生混淆。
阅读全文