Cod sursa(job #446983)

Utilizator katakunaCazacu Alexandru katakuna Data 27 aprilie 2010 11:12:15
Problema Pedefe Scor 55
Compilator cpp Status done
Runda Arhiva de probleme Marime 3.77 kb
#include <cstdio>
#include <algorithm>
using namespace std;

#define Nmax 510
#define MOD 666013

int n, m, p, Vmax, poz, val, a, b, rez;
int A[Nmax][Nmax], B[Nmax][Nmax], Sa[Nmax], Sb[Nmax], SA[4 * Nmax], SB[4 * Nmax], SA_ad[4 * Nmax], SB_ad[4 * Nmax];
int S1[Nmax], S2[Nmax], S3[Nmax];

void citire () {
	
	int i;
	
	scanf ("%d %d %d", &n, &m, &p);
	for (i = 1; i <= n; i++) {
		scanf ("%d", &S1[i]);
		if (S1[i] > Vmax) Vmax = S1[i];
	}
	
	for (i = 1; i <= m; i++) {
		scanf ("%d", &S2[i]);
		if (S2[i] > Vmax) Vmax = S2[i];
	}
	
	for (i = 1; i <= p; i++) 
		scanf ("%d", &S3[i]);
}

#define MIJ ((st + dr) >> 1)
#define N1 (nod << 1)
#define N2 ((nod << 1) + 1)

void update_A (int nod, int st, int dr) {
	
	if (st == dr) {
		SA[nod]+= val;
		if (SA[nod] >= MOD) SA[nod]-= MOD;
		return ;
	}
	
	if (poz <= MIJ) update_A (N1, st, MIJ);
	else update_A (N2, MIJ + 1, dr);
	
	SA[nod] = SA[N1] + SA[N2];
	if (SA[nod] >= MOD) SA[nod]-= MOD;
}

void update_B (int nod, int st, int dr) {
	
	if (st == dr) {
		SB[nod]+= val;
		if (SB[nod] >= MOD) SB[nod]-= MOD;
		return ;
	}
	
	if (poz <= MIJ) update_B (N1, st, MIJ);
	else update_B (N2, MIJ + 1, dr);
	
	SB[nod] = SB[N1] + SB[N2];
	if (SB[nod] >= MOD) SB[nod]-= MOD;
}

void query_A (int nod, int st, int dr) {
	
	if (a <= st && dr <= b) {
		rez+= SA[nod];
		if (rez >= MOD) rez-= MOD;
		return ;
	}
	
	if (a <= MIJ) query_A (N1, st, MIJ);
	if (MIJ < b) query_A (N2, MIJ + 1, dr);
}

void query_B (int nod, int st, int dr) {
	
	if (a <= st && dr <= b) {
		rez+= SB[nod];
		if (rez >= MOD) rez-= MOD;
		return ;
	}
	
	if (a <= MIJ) query_B (N1, st, MIJ);
	if (MIJ < b) query_B (N2, MIJ + 1, dr);
}

void dinamica () { 
	
	int i, j, k;
	
	A[0][0] = 1; 
	Sa[0] = 1;
	
	for (i = 1; i <= n; i++) {
		memset (SA, 0, sizeof (SA));
		
		poz = 0; val = 1;
		update_A (1, 0, Vmax);
		
		for (j = 1; j <= m; j++) {			
			if (S1[i] == S2[j]) {
				rez = 0; a = 0; b = S2[j];
				query_A (1, 0 ,Vmax);
				
				A[i][j]+= rez;
				if (A[i][j] >= MOD) A[i][j]-= MOD;
			}
			
			if (Sa[j]) {
				poz = S2[j]; val = Sa[j];
				update_A (1, 0, Vmax);
			}
		}
		
		for (j = 1; j <= m; j++) {
			Sa[j] = Sa[j] + A[i][j];
			if (Sa[j] >= MOD) Sa[j]-= MOD;
		}
	}
	
	memset (Sa, 0, sizeof (Sa));
	memset (SA, 0, sizeof (SA));
	Sa[0] = 1;
	
	for (k = 1; k <= p; k++) {
		
		for (i = 1; i <= n; i++) {
			memset (SA, 0, sizeof (SA));
			memset (SB, 0, sizeof (SB));
			
			poz = 0; val = Sa[0];
			if (val) update_A (1, 0, Vmax);
			
			for (j = 1; j <= m; j++) {
				if (S1[i] == S2[j]) {
					
					if (S1[i] == S3[k]) {
						rez = 0; a = 0; b = S2[j];
						query_A (1, 0 ,Vmax);
						
						B[i][j]+= rez;
						if (B[i][j] >= MOD) B[i][j]-= MOD;
					}
					
					else {
						rez = 0; a = 0; b = S2[j];
						query_B (1, 0 ,Vmax);
						
						B[i][j]+= rez;
						if (B[i][j] >= MOD) B[i][j]-= MOD;
					}
				}		
				
				poz = S2[j]; val = Sa[j];
				if (val) update_A (1, 0, Vmax);
				
				poz = S2[j]; val = Sb[j];
				if (val) update_B (1, 0, Vmax);
			}
			
			for (j = 1; j <= m; j++) { 
				Sb[j] = Sb[j] + B[i][j];
				Sa[j] = Sa[j] + A[i][j];
				if (Sa[j] >= MOD) Sa[j]-= MOD;
				if (Sb[j] >= MOD) Sb[j]-= MOD;
			}
		}
		
		memcpy (Sa, Sb, sizeof (Sa));
		memset (Sb, 0, sizeof (Sb));
		memset (Sa, 0, sizeof (Sa));
		
		for (i = 0; i <= n; i++) {
			memcpy (A[i], B[i], sizeof (A[i]));
			memset (B[i], 0, sizeof (B[i]));
		}
	}
	
	int sol = 0;
	for (i = 1; i <= n; i++)
		for (j = 1; j <= m; j++) {
			sol+= A[i][j];
			if (sol >= MOD) sol-= MOD;
		}
	
	printf ("%d", sol);
}

int main () {

	freopen ("pedefe.in", "r", stdin);
	freopen ("pedefe.out", "w", stdout);

	citire ();
	dinamica ();
	
	return 0;
}