cf 158div2 D. Black and White Tree
題意很簡單,給出一個樹中每個點相連的邊的權值和,和點的顏色,一條邊上顏色不同,求出原圖……
因為條件很少,所以必然是個spj,所以可以直接貪心,然後把最後剩餘的0邊掛到任意一點上即可
/* author:jxy lang:C/C++ university:China,Xidian University **If you need to reprint,please indicate the source** */ #include <iostream> #include <cstdio> #include <cstdlib> #include <algorithm> #include <vector> #define INF 1E9 using namespace std; #define mp make_pair #define f first #define s second vector<pair<int,int> >v[2]; int main() { int n,i,j,k,t; scanf("%d",&n); for(i=0;i<n;i++) { scanf("%d%d",&t,&k); v[t].push_back(mp(k,i+1)); } sort(v[0].begin(),v[0].end()); sort(v[1].begin(),v[1].end()); for(i=j=0;i<v[0].size()&&j<v[1].size();) { k=min(v[0][i].f,v[1][j].f); printf("%d %d %d\n",v[0][i].s,v[1][j].s,k); v[0][i].f-=k;v[1][j].f-=k; if(v[0][i].f)j++; else if(v[1][j].f)i++; else if(i<v[0].size()-1)i++; else j++; } return 0; }
最後更新:2017-04-04 07:03:16