Submission #607603

#	Time	Username	Problem	Language	Result	Execution time	Memory
607603		Pietra	Lottery (CEOI18_lot)	C++14	65 / 100	1992 ms	65536 KiB

lot

#include<bits/stdc++.h>
#define ll long long
using namespace std ; 

const int prime = 37 ; 
const int mod = 1e9 + 7 ; 
const int maxn = 1e4 + 5 ; 

int n, l, v[maxn], dist[maxn][maxn], k, q ;  
ll pot[maxn], inv[maxn], pref[maxn], val[maxn] ; 

void solve(){
	
	for(int i = 1 ; i <= n ; i++){
		for(int j = 1 ; j <= n ; j++){ // dist de i p j 
			if(i == j || i + l - 1 > n || j + l - 1 > n) continue ; 
			int ctr = 0 ; 
			for(int k = 0 ; k < l ; k++) if(v[i+k] != v[j+k]) ctr++ ; 
			dist[i][j] = dist[j][i] = ctr ; 
		}
	}

}

ll exp(ll a, ll b){
	ll ans = 1 ; 
	while(b){
		if(b&1) ans = (ans*a)%mod ; 
		a = (a*a)%mod ; 
		b >>= 1 ; 
	}
	return ans ; 
}

void solve_hash(){

	pot[0] = 1LL ; 

	for(int i = 1 ; i < maxn ; i++) pot[i] = (pot[i-1]*prime)%mod ; 

	inv[0] = 1LL ; inv[1] = exp(prime, mod-2) ; 
    
    for(int i = 2 ; i < maxn ; i++) inv[i] = (inv[i-1]*inv[1])%mod ; 

	for(int i = 1 ; i <= n ; i++) pref[i] = (pref[i-1] + (v[i]*pot[i]*1LL)%mod)%mod ;

	vector<ll> vals ; 

	for(int i = 1 ; i + l - 1 <= n ; i++){ 
		ll v1 = pref[i+l-1] ; 
		if(i - 1 >= 1) v1 = (v1 - pref[i-1] + mod)%mod ; 
		v1 = (v1*inv[i])%mod ; 
		vals.push_back(v1) ; val[i] = v1 ; 
	}

	sort(vals.begin(), vals.end()) ; 

	for(int i = 1 ; i + l - 1 <= n ; i++){
		int l =  lower_bound(vals.begin(), vals.end(), val[i]) - vals.begin() ; 
		int r =  upper_bound(vals.begin(), vals.end(), val[i]) - vals.begin() ;
		cout << (r - l - 1) << " " ; 
	}

	cout << "\n" ; 

}

int main(){

	ios_base::sync_with_stdio(false) ; cin.tie(NULL) ; 

	cin >> n >> l ; 

	for(int i = 1 ; i <= n ; i++) cin >> v[i] ;

	// for(int i = 1  ; i <= n ; i++){
	// 	for(int j = 1 ; j <= n ; j++){
	// 		cout << dist[i][j] <<" " ; 
	// 	}
	// 	cout << "\n" ; 
	// }   

    cin >> q ; 
    cin >> k ; 
    
    if(q == 1 && !k){
    	solve_hash() ;
    	exit(0) ; 
    } 

    solve() ;
    
    for(int i = 1 ; i + l - 1 <= n ; i++){
       	int ct = 0 ; 
      	for(int j = 1 ; j + l - 1 <= n ; j++) if(i != j && dist[i][j] <= k) ct++ ; 
      	cout << ct << " " ; 
    }
    cout << "\n" ;

    for(int i = 1 ; i < q ; i++){
    	cin >> k ; 
    	for(int i = 1 ; i + l - 1 <= n ; i++){
         	int ct = 0 ; 
         	for(int j = 1 ; j + l - 1 <= n ; j++) if(i != j && dist[i][j] <= k) ct++ ; 
         	cout << ct << " " ; 
        }
        cout << "\n" ; 
    }

}

• Subtask 1: 25 / 25
• Subtask 2: 20 / 20
• Subtask 3: 20 / 20
• Subtask 4: 0 / 15
• Subtask 5: 0 / 20