Поменять LexOrder Union Найти
Итак, я задаю вопросы для интервью и наткнулся на этот вопрос: учитывая строку str и массив пар, которые указывают, какие индексы в строке можно поменять местами, верните самую большую лексикографически строку, полученную в результате выполнения разрешенных перестановок. Вы можете поменять индексы любое количество раз.
пример
Для str = "abdc" и pair = [[1, 4], [3, 4]] выходные данные должны быть swapLexOrder(str, pair) = "dbca".
Меняя местами указанные индексы, вы получаете строки: "cbda", "cbad", "dbac", "dbca". Самая большая лексикографическая строка в этом списке - "dbca".
У меня есть рабочий ответ, связанный с поиском профсоюзов, но мой ответ слишком медленный:
[time limit] 4000ms (js)
0 ≤ pairs.length ≤ 5000,
pairs[i].length = 2.
1 ≤ str.length ≤ 10^4
Может ли кто-нибудь помочь мне настроить мой код для увеличения его скорости? Вот код, который я имею:
function swapLexOrder(str, pairs) {
if (!pairs.length){
return str
}
answerHash = {}
unmoved = findUnmoved(pairs, str.length)
unionsArr = findUnions(pairs)
for (i in unmoved){
answerHash[unmoved[i]] = str[(unmoved[i]-1)]
}
unionsArr.forEach(function(union){
letters = []
for (i in union){
letters.push(str[(union[i]-1)])
}
letters.sort()
union.sort(function(a,b){
return b-a
})
for (j in union){
answerHash[union[j]] = letters[j]
}
})
string = []
for (keys in answerHash){
string.push(answerHash[keys])
}
return string.join('')
}
//if two pairs share a number they belong in the same array
findUnions = function(pairs, unions){
if (!unions){
unions = [pairs[0]];
pairs.shift();
}else{
if(pairs.length){
unions.push(pairs[0])
pairs.shift()
}
}
if (!pairs.length){
return unions
}
unite = true
while (unite && pairs.length){
unite = false
loop1:
for (i in unions){
loop2:
for (j in pairs){
if (unions[i].includes(pairs[j][0])){
unions[i].push(pairs[j][1])
pairs.splice(j, 1)
unite = true
break loop1
}else if (unions[i].includes(pairs[j][1])){
unions[i].push(pairs[j][0])
pairs.splice(j, 1)
unite = true
break loop1
}
}
}
}
return findUnions(pairs, unions)
}
findUnmoved = function(pairs, length){
range = []
for (var i=1;i<length+1;i++){
range.push(i);
}
allNum = [].concat.apply([], pairs)
range = range.filter(function(x){
return (!allNum.includes(x))
})
return range
}
Вероятно, это функция, которую я использую, чтобы найти профсоюзы, но я подумал, может, я смогу сделать это без создания хеша? Кроме того, если вы узнаете о лучшем методе решения проблемы, я всегда готов учиться чему-то новому. Спасибо!
1 ответ
Этот работает быстрее.
function swapLexOrder(str, pairs) {
//Turn pairs into edge lists: O(n+m)
var graph = new Array(str.length).fill(0).map(e=>[]);
for(var pair of pairs) {
graph[pair[0]-1].push(pair[1]-1);
graph[pair[1]-1].push(pair[0]-1);
}
//Build all the ccs with dfs: O(n+m)
var ccs = [], ccnum = 0;
for(var c in str) {
if(ccs[c])
continue;
ccs[c] = ++ccnum;
var dfs = [...graph[c]];
while(dfs.length) {
var d = dfs.shift();
if(ccs[d])
continue;
ccs[d] = ccnum;
dfs.push(...graph[d]);
}
}
//Group words by ccs: O(n)
var ccWords = new Array(ccnum).fill(0).map(e=>[]);
for(var c in str) {
ccWords[ccs[c]-1].push(str[c]);
}
//Sort all words: O(n log n)
ccWords.map(e=>e.sort());
//Build the new string: O(n)
var output = "";
for(var c in str) {output += ccWords[ccs[c]-1].pop(); }
return output;
}