#include<vector>
long long int calculate_diamonds (int N, int M, int K, std::vector <int> u, std::vector <int> v, std::vector <int> d);