tookunn’s diary

主に競技プログラミング関係

RUPC2016 Day1 B ハミング距離 / Hamming Distance

考察

ハミング距離がDである値の中で最大のものを求めなければならないので、Xを左から見ていき,'0'である文字をD個'1'に変える。
もし、'0'である文字がD個に満たない場合はXを右から見ていき,先程'0'から'1'に変えていない文字なおかつ'1'である文字を'0'に変えていく。


ソースコード

import java.io.IOException;
import java.io.InputStream;
import java.io.PrintWriter;
import java.util.NoSuchElementException;

public class Main {
	int N,D;
	String X;
	public void solve() {
		N = nextInt();
		X = next();
		D = nextInt();

		char[] ch = X.toCharArray();
		boolean[] used = new boolean[N];
		int change = 0;
		for(int i = 0;i < N;i++){
			if(change >= D)break;
			if(ch[i] == '0'){
				used[i] = true;
				change++;
				ch[i] = '1';
			}
		}

		for(int i = N - 1;i >= 0;i--){
			if(change >= D)break;
			if(!used[i] && ch[i] == '1'){
				change++;
				ch[i] = '0';
			}
		}

		out.println(String.valueOf(ch));
	}
	public static void main(String[] args) {
		out.flush();
		new Main().solve();
		out.close();
	}

	/* Input */
	private static final InputStream in = System.in;
	private static final PrintWriter out = new PrintWriter(System.out);
	private final byte[] buffer = new byte[2048];
	private int p = 0;
	private int buflen = 0;

	private boolean hasNextByte() {
		if (p < buflen)
			return true;
		p = 0;
		try {
			buflen = in.read(buffer);
		} catch (IOException e) {
			e.printStackTrace();
		}
		if (buflen <= 0)
			return false;
		return true;
	}

	public boolean hasNext() {
		while (hasNextByte() && !isPrint(buffer[p])) {
			p++;
		}
		return hasNextByte();
	}

	private boolean isPrint(int ch) {
		if (ch >= '!' && ch <= '~')
			return true;
		return false;
	}

	private int nextByte() {
		if (!hasNextByte())
			return -1;
		return buffer[p++];
	}

	public String next() {
		if (!hasNext())
			throw new NoSuchElementException();
		StringBuilder sb = new StringBuilder();
		int b = -1;
		while (isPrint((b = nextByte()))) {
			sb.appendCodePoint(b);
		}
		return sb.toString();
	}

	public int nextInt() {
		return Integer.parseInt(next());
	}

	public long nextLong() {
		return Long.parseLong(next());
	}

	public double nextDouble() {
		return Double.parseDouble(next());
	}
}