playfair

class Solution:
    def Encode(self , key , str ):
        # write code here
        uniquekey =[]
        for x in key:
            if x =='j':
                x = 'i'                
            if x not in uniquekey:
                uniquekey.append(x)
        table =['']*26
        for i in range(0,26):
            table[i] = chr(ord('a')+ i)
        table.remove('j')
        for x in uniquekey:
            table.remove(x)
        for x in table:
            uniquekey.append(x)
        playfair = [['']*5]*5
        k=0
        while k<25:
            for i in range(0,5):
                tmp=[]
                for j in range(0,5):
                    tmp.append(uniquekey[k])
                    k+=1
                playfair[i]=tmp
        s11=''
        s2=''
        strlen = len(str)
        if strlen%2!=0:
            s11=str[0:-1]
            s2=str[-1]
        else:
            s11=str
        s1=''
        for x in s11:
            if x!='j':
                s1+=x
            else:
                s1+='i'
        tmp=0
        res=''
        while tmp<len(s1)-1:
            if s1[tmp]==s1[tmp+1]:
                res+=s1[tmp]
                res+=s1[tmp+1]
                tmp+=2
                continue
            x1,x2,y1,y2=0,0,0,0
            for i in range(0,5):
                for j in range(0,5):
                    if playfair[i][j]==s1[tmp]:
                        x1=i
                        y1=j
                    elif playfair[i][j]==s1[tmp+1]:
                        x2=i
                        y2=j
            if x1==x2:
                if y1==4:
                    y1=-1
                if y2==4:
                    y2=-1
                res+=playfair[x1][y1+1]
                res+=playfair[x1][y2+1]
                tmp+=2
            elif y1==y2:
                if x1==4:
                    x1=-1
                if x2==4:
                    x2=-1
                res+=playfair[x1+1][y1]
                res+=playfair[x2+1][y1]
                tmp+=2
            else:
                res+=playfair[x1][y2]
                res+=playfair[x2][y1]
                tmp+=2
        res+=s2
        return res

 public String Encode (String key, String str) {
        str=str.replace("j","i");
        key=key.replace("j","i");
        Map<Character,Integer> map=new HashMap<>();
        List<Character> list=new ArrayList<>();
        int index=0;
        for(int i=0;i<key.length();i++) {
        	char c=key.charAt(i);
        	if(!map.containsKey(c)) {
        		map.put(c,index++);
        		list.add(c);
        	}
        }
        for(int i=0;i<26;i++) {
        	char c=(char)('a'+i);
        	if(c=='j')
                continue;
        	if(!map.containsKey(c)) {
        		map.put(c,index++);
        		list.add(c);
        	}
        }
        StringBuilder sb=new StringBuilder();
        for(int i=0;i<str.length();i+=2) {
        	char c1=str.charAt(i);
        	if(i==str.length()-1) {
        		sb.append(c1);
        		break;
        	}
        	char c2=str.charAt(i+1);
        	if(c1==c2) {
        		sb.append(c1).append(c1);
        	}else {
        		getFair(map,list,c1,c2,sb);
        	}
        }
		return sb.toString();
    }
    private  void getFair(Map<Character,Integer> map, List<Character> list,char c1,char c2,StringBuilder sb) {
		int n1=map.get(c1);
		int n2=map.get(c2);
		int x1=n1/5;
		int y1=n1%5;
		int x2=n2/5;
		int y2=n2%5;
		
		if(x1==x2) {
			y1=(++y1==5?0:y1);
			y2=(++y2==5?0:y2);
		}else if(y1==y2) {
			x1=(++x1==5?0:x1);
			x2=(++x2==5?0:x2);
		}else {
			int tmp=y1;
			y1=y2;
			y2=tmp;
		}
		n1=x1*5+y1;
		n2=x2*5+y2;
		sb.append(list.get(n1)).append(list.get(n2));
	}