Cod sursa(job #2340712)

Utilizator AxellApostolescu Alexandru Axell Data 10 februarie 2019 21:04:36
Problema Ciurul lui Eratosthenes Scor 100
Compilator c-64 Status done
Runda Arhiva educationala Marime 1.47 kb
#include <stdio.h>
#include <stdlib.h>
typedef unsigned char *set;

// Se va folosi convetia bit = 1 => bit nu e in set
// Daca bitul i din byteul b este in set => 16 * b + 2 * i + 1 este prim
void remove_from_set(set s, int val) {
	if (!(val & 1)) {
		return;
	}
	int byte = (val >> 4), bit = ((val >> 1) % 8);
	unsigned char mask = 1 << bit;
	s[byte] |= mask;
}

int is_in_set(set s, int val) {
	// Cazuri speciale
	if (val == 1) {
		return 0;
	}
	if (val == 2) {
		return 1;
	}
	if (!(val & 1)) {
		return 0;
	}
	int byte = (val >> 4), bit = ((val >> 1) % 8);
	unsigned char mask = 1 << bit;
	if ((s[byte] & mask)) {
		return 0;
	}
	return 1;
}

// Contorul incepe de la 1 deoarece 2 este in set
int ciur(set s, int n) {
	int i, ct = 1, j;
	for (i = 1 ; ((i << 1) + 1) <= n ; ++i) {
		if (is_in_set(s, ((i << 1) + 1))) {
			ct++;
			for (j = (((i << 1) + 1) << 1); j <= n ; j += ((i << 1) + 1)) {
				remove_from_set(s, j);
			}
		}
	}
	return ct;
}

int main() {
	FILE *in, *out;
	if (((in = fopen("ciur.in", "rt")) == NULL)) {
		printf("Nu am putut deschide fisierul de input!");
		return -1;
	}
	if (((out = fopen("ciur.out", "wt")) == NULL)) {
		printf("Nu am putut deschide fisierul de output!");
		return -2;
	}
	int n;
	fscanf(in, "%d", &n);
	set s = calloc((n + 15) / 16, sizeof(char));
	if (s == NULL) {
		printf("Nu am putut aloca memorie\n");
		return -3;
	}
	int result = ciur(s, n);
	fprintf(out, "%d\n", result);
	// Freeing the memory
	free(s);
	fclose(in);
	fclose(out);
	return 0;
}