求助 LOJ AC,但 Luogu WA
查看原帖
求助 LOJ AC,但 Luogu WA
201007
Leasier楼主2022/10/25 20:03

RT

代码:

#include <stdio.h>
#include <stdbool.h>

int sum[100007][7];
char s[100007];

int main(){
	int t;
	scanf("%d", &t);
	for (int i = 1; i <= t; i++){
		int n;
		bool flag = true;
		scanf("%d", &n);
		scanf("%s", &s[1]);
		for (int j = 1; j <= n; j++){
			if (s[j] != '0'){
				flag = false;
				break;
			}
		}
		if (flag){
			for (int j = n; j >= 1; j--){
				printf("%d ", j);
			}
			printf("\n");
			continue;
		}
		int m = (n - 1) / 2;
		for (int j = 1; j <= n; j++){
			sum[j][0] = sum[j - 1][0] + (s[j] == '0' ? 1 : 0);
			sum[j][1] = sum[j - 1][1] + (s[j] == '1' ? 1 : 0);
		}
		for (int j = 1; j <= n; j++){
			if (sum[j - 1][0] + (sum[n][1] - sum[j][1]) == m){
				for (int k = 1; k < j; k++){
					if (s[k] == '1') printf("%d ", k);
				}
				for (int k = j + 1; k <= n; k++){
					if (s[k] == '0') printf("%d ", k);
				}
				printf("%d ", j);
				for (int k = 1; k < j; k++){
					if (s[k] == '0') printf("%d ", k);
				}
				for (int k = j + 1; k <= n; k++){
					if (s[k] == '1') printf("%d ", k);
				}
				printf("\n");
				break;
			}
		}
	}
	return 0;
}
2022/10/25 20:03
加载中...