LeetCode 1061. 按字典序排列最小的等效字符串(並查集)

文章目錄

1. 題目

給出長度相同的兩個字符串:A 和 B,其中 A[i] 和 B[i] 是一組等價字符。
舉個例子,如果 A = "abc" 且 B = "cde",那麼就有 'a' == 'c', 'b' == 'd', 'c' == 'e'

等價字符遵循任何等價關係的一般規則:

自反性:'a' == 'a'
對稱性:'a' == 'b' 則必定有 'b' == 'a'
傳遞性:'a' == 'b''b' == 'c' 就表明 'a' == 'c'
例如,A 和 B 的等價信息和之前的例子一樣,
	那麼 S = "eed", "acd""aab",這三個字符串都是等價的,
	而 "aab" 是 S 的按字典序最小的等價字符串

利用 A 和 B 的等價信息,找出並返回 S 的按字典序排列最小的等價字符串。

示例 1:
輸入:A = "parker", B = "morris", S = "parser"
輸出:"makkek"
解釋:根據 A 和 B 中的等價信息,
我們可以將這些字符分爲 [m,p], [a,o], [k,r,s], [e,i]4 組。
每組中的字符都是等價的,並按字典序排列。所以答案是 "makkek"。

示例 2:
輸入:A = "hello", B = "world", S = "hold"
輸出:"hdld"
解釋:根據 A 和 B 中的等價信息,
我們可以將這些字符分爲 [h,w], [d,e,o], [l,r]3 組。
所以只有 S 中的第二個字符 'o' 變成 'd',最後答案爲 "hdld"。

示例 3:
輸入:A = "leetcode", B = "programs", S = "sourcecode"
輸出:"aauaaaaada"
解釋:我們可以把 A 和 B 中的等價字符分爲 
[a,o,e,r,s,c], [l,p], [g,t][d,m]4 組,
因此 S 中除了 'u''d' 之外的所有字母都轉化成了 'a',最後答案爲 "aauaaaaada"。
 
提示:
字符串 A,B 和 S 僅有從 'a''z' 的小寫英文字母組成。
字符串 A,B 和 S 的長度在 11000 之間。
字符串 A 和 B 長度相同。

來源:力扣(LeetCode)
鏈接:https://leetcode-cn.com/problems/lexicographically-smallest-equivalent-string
著作權歸領釦網絡所有。商業轉載請聯繫官方授權,非商業轉載請註明出處。

2. 解題

請參考:數據結構–並查集(Disjoint-Set)

  • 並查集merge的時候,讓祖先字符更小的作爲代表
class dsu
{
	vector<int> f;
public:
	dsu(int n)
	{
		f.resize(n);
		for(int i = 0; i < n; ++i)
			f[i] = i;
	}
	void merge(int a, int b)
	{
		int fa = find(a), fb = find(b);
        if(fa > fb)//字母小的當代表,關鍵點
		    f[fa] = fb;
        else
            f[fb] = fa;
	}
	int find(int a)
	{
		int origin = a;
		while(a != f[a])
			a = f[a];
		return f[origin] = a;
	}
};
class Solution {
public:
    string smallestEquivalentString(string A, string B, string S) {
    	dsu u(26);
    	for(int i = 0; i < A.size(); ++i)
    		u.merge(A[i]-'a', B[i]-'a');
    	for(int i = 0; i < S.size(); ++i)
    		S[i] = u.find(S[i]-'a')+'a';
    	return S;
    }
};

0 ms 6.7 MB

  • 一開始沒想到字符小的作爲代表,是用set做了下處理,代碼如下
class dsu
{
public:
	vector<int> f;
	dsu(int n)
	{
		f.resize(n);
		for(int i = 0; i < n; ++i)
			f[i] = i;
	}
	void merge(int a, int b)
	{
		int fa = find(a), fb = find(b);
		f[fa] = fb;
	}
	int find(int a)
	{
		int origin = a;
		while(a != f[a])
			a = f[a];
		return f[origin] = a;
	}
};
class Solution {
public:
    string smallestEquivalentString(string A, string B, string S) {
    	dsu u(26);
    	for(int i = 0; i < A.size(); ++i)
    		u.merge(A[i]-'a', B[i]-'a');
        unordered_map<int, set<int>> m;
        for(int i = 0; i < 26; ++i)
            m[u.find(i)].insert(i);//分組,set內 begin最小,作爲代表
        for(int i = 0; i < 26; ++i)
        	u.f[i] = *m[u.f[i]].begin();//更新祖先爲集團內最小的字符
    	for(int i = 0; i < S.size(); ++i)
    		S[i] = u.find(S[i]-'a')+'a';
    	return S;
    }
};

8 ms 7.4 MB


我的CSDN博客地址 https://michael.blog.csdn.net/

長按或掃碼關注我的公衆號(Michael阿明),一起加油、一起學習進步!
Michael阿明

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章