831. 隐藏个人信息

831. 隐藏个人信息

 

 

class Solution {
public:
	string maskPII(string S)
	{
		int pos = S.find('@');
		if (pos!=string::npos)
		{
			return Email(S, pos);
		}
		else
		{
			return PhoneNumbers(S);
		}
	}
	string Email(string s,int len)
	{
		string ans;
		if (s[0] >= 'A'&&s[0] <= 'Z')
		{
			s[0] += 32;
		}
		ans += s[0];
		ans += "*****";
		if (s[len - 1] >= 'A'&&s[len - 1] <= 'Z')
		{
			s[len - 1] += 32;
		}
		ans += s[len - 1];
		ans += '@';
		for (int i = len+1; i < s.length(); i++)
		{
			if (s[i]>='A'&&s[i]<='Z')
			{
				s[i] += 32;
			}
			ans += s[i];
		}
		return ans;
	}
	string PhoneNumbers(string s)
	{
		string ans;
		int count = 0;
		string ans13= "+***-***-***-";
		string ans12 = "+**-***-***-";
		string ans11= "+*-***-***-";
		string ans2 = "***-***-";
		for (int i = 0; i < s.length(); i++)
		{
			if (s[i]>='0'&&s[i]<='9')
			{
				ans += s[i];
			}
		}
		if (ans.length()==13)
		{
			return ans13 += ans.substr(ans.size() - 4);

		}
		if (ans.length()==12)
		{
			return ans12 += ans.substr(ans.size() - 4);
		}
		if (ans.length()==11)
		{
			return ans11 += ans.substr(ans.size() - 4);
		}
		else
		{
			return ans2 += ans.substr(ans.size() - 4);
		}
	}
};