Online Judge | Problem Set | Authors | Online Contests | User | ||||||
---|---|---|---|---|---|---|---|---|---|---|
Web Board Home Page F.A.Qs Statistical Charts | Current Contest Past Contests Scheduled Contests Award Contest |
牛客上过了,poj哇了,不知道哪里卡了#include <iostream> #include <cmath> #include <cstdio> #include <cstring> #include <string> #include <map> #include <iomanip> #include <algorithm> #include <queue> #include <stack> #include <set> #include <vector> // #include <bits/stdc++.h> #define fastio ios_base::sync_with_stdio(0);cin.tie(0);cout.tie(0); #define sp ' ' #define endl '\n' #define FOR(i,a,b) for( int i = a;i <= b;++i) #define bug cout<<"--------------"<<endl #define P pair<int, int> #define fi first #define se second #define pb(x) push_back(x) #define ppb() pop_back() #define mp(a,b) make_pair(a,b) #define ms(v,x) memset(v,x,sizeof(v)) #define rep(i,a,b) for(int i=a;i<=b;i++) #define repd(i,a,b) for(int i=a;i>=b;i--) #define sca3(a,b,c) scanf("%d %d %d",&(a),&(b),&(c)) #define sca2(a,b) scanf("%d %d",&(a),&(b)) #define sca(a) scanf("%d",&(a)); #define sca3ll(a,b,c) scanf("%lld %lld %lld",&(a),&(b),&(c)) #define sca2ll(a,b) scanf("%lld %lld",&(a),&(b)) #define scall(a) scanf("%lld",&(a)); using namespace std; typedef long long ll; ll gcd(ll a,ll b){return b?gcd(b,a%b):a;} ll lcm(ll a,ll b){return a/gcd(a,b)*b;} ll powmod(ll a, ll b, ll mod){ll sum = 1;while (b) {if (b & 1) {sum = (sum * a) % mod;b--;}b /= 2;a = a * a % mod;}return sum;} const double Pi = acos(-1.0); const double epsilon = Pi/180.0; const int maxn = 7e5+10; int tot = 0; int head[maxn],ver[maxn],edge[maxn],nexts[maxn]; void add(int x,int y,int z) { ver[++tot] = y,edge[tot] = z,nexts[tot] = head[x] , head[x] = tot; } void add(int x,int y){ ver[++tot] = y; nexts[tot] = head[x]; head[x] = tot; //edge[tot] = z; } int k,n,m; ll d[maxn],f[maxn][30],vis[maxn],cf[maxn]; ll sum[maxn]; void bfs(){ queue<int> que; que.push(1); d[1] = 1; while(!que.empty()){ int x = que.front(); que.pop(); for(int i = head[x]; i; i = nexts[i]){ int y = ver[i]; if(d[y]) continue; //dist[y] = dist[x] + edge[i]; d[y] = d[x] + 1; f[y][0] = x; for(int j = 1; j <= k;j++){ f[y][j] = f[f[y][j-1]][j-1]; } que.push(y); } } } int lca(int x, int y){ if(d[x] > d[y]) swap(x,y); //设y比较深,深度d大 for(int i = k;i >= 0;i--){ if(d[f[y][i]] >= d[x]) y = f[y][i]; } if(x == y) return x; for(int i = k;i >= 0;i--){ if(f[x][i] != f[y][i]){ x = f[x][i]; y = f[y][i]; } } return f[x][0]; } ll dfs(int x) { ll cnt = 0; for(int i = head[x];i;i = nexts[i]){ int y = ver[i]; if(vis[y] == 1) continue; vis[y] = 1; cnt += dfs(y); } return sum[x] = cnt+cf[x]; } int main() { //freopen("input.txt", "r", stdin); sca2(n,m); //cin>>n>>m; rep(i,1,n-1){ int x,y; sca2(x,y); //cin>>x>>y; add(x,y); } k = (int)(log(n) / log(2)) + 1; bfs(); rep(i,1,m){ int x,y; sca2(x,y); cf[x]++; cf[y]++; cf[lca(x,y)] -= 2; } vis[1] = 1; dfs(1); ll ans = 0; rep(i,2,n){ if(sum[i] == 0) ans += m; else if(sum[i] == 1) ans ++; } printf("%lld\n",ans ); //cout<<ans<<endl; } Followed by: Post your reply here: |
All Rights Reserved 2003-2013 Ying Fuchen,Xu Pengcheng,Xie Di
Any problem, Please Contact Administrator