输入使用cin会超时,改为用scanf
此题有坑,不能直接判断输入中有两个相同地址,因为可能会有没用的输入数据,需要使用栈或者链表生成word再从尾部输出。
#define _CRT_SECURE_NO_WARNINGS
#include<string>#include<string.h>
#include<vector>
#include<map>
#include<stack>
#include<iostream>
#include<algorithm>
#include<stdio.h>
using namespace std;
int main()
{
int A, B, count;
cin >> A >> B >> count;
if (A == B)
{
if (A == -1)
cout << "-1\n";
else
{
printf("%05d\n", A);
}
return 0;
}
map<int, int> mm;
for (int i = 0; i < count; i++)
{
int Address, Next;
char Data;
scanf("%d %c %d", &Address, &Data, &Next);
mm[Address] = Next;
}
stack<int> s_A, s_B;
s_A.push(A);
while (A != -1)
{
s_A.push(mm[A]);
A = mm[A];
}
s_B.push(B);
while (B != -1)
{
s_B.push(mm[B]);
B = mm[B];
}
int iOut = -1;
while (s_A.size()>0&&s_B.size()>0&&s_A.top() == s_B.top())
{
iOut = s_A.top();
s_A.pop(); s_B.pop();
}
if (iOut == -1)
cout << "-1\n";
else
printf("%05d\n",iOut);
return 0;
}