3226774_ac_6141ms_7496k.java

北大大牛代码 1240道题的原代码 超级权威

import java.io.*;
import java.util.*;

public class Main{

    private void solve() throws IOException {
		BufferedReader in = new BufferedReader (new InputStreamReader (System.in));

        String t1 = in.readLine();
        String s1 = in.readLine();
        int n = s1.length();
        BitSet ss = stringToBitSet(s1 + s1);
        BitSet t = stringToBitSet(t1);
        HashSet<BitSet> shifts = new HashSet<BitSet>();
        for (int i = 0; i < n; i++) {
            BitSet sh = ss.get(i, i + n);
            shifts.add(sh);
        }
        boolean ok = false;
        for (BitSet sh : shifts) {
            BitSet st = (BitSet) sh.clone();
            st.xor(t);
            if (shifts.contains(st)) {
                ok = true;
                break;
            }
        }
        if (ok)
            System.out.println("Yes");
        else
            System.out.println("No");
    }

    private BitSet stringToBitSet(String s) {
        BitSet ret = new BitSet(s.length());
        for (int i = 0; i < s.length(); i++)
		{
			if (s.charAt(i) == '1')
			{
				ret.set(i);
			}
		}
        return ret;
    }

    public static void main(String[] args) throws IOException{
        new Main().solve();
    }
}