给定一个由字符‘N’、’S’、’E’和‘W’组成的字符串路径,分别表示在北、南、东和西方向上的 1 个单位移动,任务是找到完成整个行程所需的时间。从原点开始的路径,如果分别需要 2 分钟和 1 分钟才能在未访问和访问过的路段上行驶。
例子 :
Input: path = “NNES”
Output : 8
Explanation: Since every segment is visited only once, cost = 2 * 4 = 8.
Input : path = “NSE”
Output : 5
Explanation:
Step 1: Travel north. Time Taken = 2 minutes.
Step 2: Travel south on that same visited segment. Time Taken = 1 minutes.
Step 3: Travel east.Time Taken = 2 minutes. Therefore, total time taken = 2 + 1 + 2 = 5.
方法:想法是使用一个 Set 来存储所有访问过的段,在访问每个段之前,检查它是否存在于 Set 中。请按照以下步骤解决问题。
- 初始化一个集合s来存储一对整数。该集合将存储所有访问过的段。
- 初始化两个整数x = 0 和y = 0 表示当前位置。此外,初始化一个变量time = 0 以存储穿越完整路径所需的总时间。
- 遍历字符串并按照以下步骤操作
- 将两个整数p和q分别初始化为x和y 。
- 如果path[i]等于‘N’增加y ,否则如果path[i]等于‘S’减少y,否则如果path[i]等于‘E’增加x ,否则减少x 。
- 检查段 { p + x , q + y } 是否存在于集合中。如果它确实将时间值加 1,否则将时间值加 2 。
- 将段 { p + x , q + y } 插入到集合中。
- 完成上述步骤后,打印时间值。
下面是上述方法的实现。
C++
// C++ code for the above approach
#include
using namespace std;
// Function to calculate time
// taken to travel the path
void calcTotalTime(string path)
{
// Stores total time
int time = 0;
// Initial position
int x = 0, y = 0;
// Stores visited segments
set > s;
for (int i = 0; i < path.size(); i++) {
int p = x;
int q = y;
if (path[i] == 'N')
y++;
else if (path[i] == 'S')
y--;
else if (path[i] == 'E')
x++;
else if (path[i] == 'W')
x--;
// Check whether segment
// is present in the set
if (s.find({ p + x, q + y })
== s.end()) {
// Increment the value
// of time by 2
time += 2;
// Insert segment into the set
s.insert({ p + x, q + y });
}
else
time += 1;
}
// Print the value
// of time
cout << time << endl;
}
// Driver Code
int main()
{
string path = "NSE";
calcTotalTime(path);
return 0;
}
输出:
5
时间复杂度: O(NlogN)
辅助空间: O(N)