-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathChefAndDigitJumps.cpp
57 lines (53 loc) · 1.22 KB
/
ChefAndDigitJumps.cpp
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
/// Chef And Digit Jumps : Problem Code: DIGJUMP : Codechef
#include <bits/stdc++.h>
using namespace std;
void solve()
{
string s;
cin>>s;
int n=s.size();
queue<int> q;
vector<int> dist(n+1,0);
vector<int> vis(n+1,0);
vector<int> adj[11];
for(int i=0;i<n;i++)
{
int val = (int) s[i]-'0';
adj[val].push_back(i);
}
int start = 0;
vis[0]=1;
q.push(start);
while(!q.empty())
{
int ind = q.front();
q.pop();
if(ind==n-1) break;
int val = s[ind]-'0';
for(int i=0;i<adj[val].size();i++)
{
if(vis[adj[val][i]]) continue;
vis[adj[val][i]]=1;
dist[adj[val][i]] = dist[ind]+1;
q.push(adj[val][i]);
}
adj[val].clear();
if(ind-1>=0 && ind-1<n && vis[ind-1]==0)
{
vis[ind-1]=1;
q.push(ind-1);
dist[ind-1]=dist[ind]+1;
}
if(ind+1>=0 && ind+1<n && vis[ind+1]==0)
{
vis[ind+1]=1;
q.push(ind+1);
dist[ind+1]=dist[ind]+1;
}
}
cout<<dist[n-1]<<endl;
}
int main() {
solve();
return 0;
}