zoukankan      html  css  js  c++  java
  • 1244D

    1244D - Paint the Tree

    #include <iostream>
    #include <vector>
    #include <algorithm>
    #include <string>
    #include <set>
    #include <queue>
    #include <map>
    #include <sstream>
    #include <cstdio>
    #include <cstring>
    #include <numeric>
    #include <cmath>
    #include <iomanip>
    #include <deque>
    #include <bitset>
    //#include <unordered_set>
    //#include <unordered_map>
    #define ll              long long
    #define pii             pair<int, int>
    #define rep(i,a,b)      for(ll  i=a;i<=b;i++)
    #define dec(i,a,b)      for(ll  i=a;i>=b;i--)
    #define forn(i, n)      for(ll i = 0; i < int(n); i++)
    using namespace std;
    int dir[4][2] = { { 1,0 },{ 0,1 } ,{ 0,-1 },{ -1,0 } };
    const long long INF = 0x7f7f7f7f7f7f7f7f;
    const int inf = 0x3f3f3f3f;
    const double pi = 3.14159265358979323846;
    const double eps = 1e-6;
    const int mod = 998244353;
    const int N = 1e6 + 5;
    //if(x<0 || x>=r || y<0 || y>=c)
    
    inline ll read()
    {
        ll x = 0; bool f = true; char c = getchar();
        while (c < '0' || c > '9') { if (c == '-') f = false; c = getchar(); }
        while (c >= '0' && c <= '9') x = (x << 1) + (x << 3) + (c ^ 48), c = getchar();
        return f ? x : -x;
    }
    ll gcd(ll m, ll n)
    {
        return n == 0 ? m : gcd(n, m % n);
    }
    ll lcm(ll m, ll n)
    {
        return m * n / gcd(m, n);
    }
    bool prime(int x) {
        if (x < 2) return false;
        for (int i = 2; i * i <= x; ++i) {
            if (x % i == 0) return false;
        }
        return true;
    }
    inline int qpow(int x, ll n) {
        int r = 1;
        while (n > 0) {
            if (n & 1) r = 1ll * r * x % mod;
            n >>= 1; x = 1ll * x * x % mod;
        }
        return r;
    }
    inline int add(int x, int y) {
        return ((x%mod)+(y%mod))%mod;
    }
    inline int sub(int x, int y) {
        x -= y;
        return x < 0 ? x += mod : x;
    }
    inline int mul(int x, int y) {
        return (1ll * (x %mod) * (y % mod))%mod;
    }
    inline int Inv(int x) {
        return qpow(x, mod - 2);
    }
    
    int main()
    {
        int n;
        cin >> n;
        vector<vector<int>> c(3, vector<int>(n + 1)), g(n + 1);
        rep(i, 0, 2)
        {
            rep(j, 1, n)
                cin >> c[i][j];
        }
        rep(i, 1, n - 1)
        {
            int u, v;
            cin >> u >> v;
            g[u].push_back(v);
            g[v].push_back(u);
        }
        int fg = 0,head;
        vector<int> v1;
        rep(i, 1, n)
        {
            if (g[i].size() > 2)
            {
                cout << -1 << endl;
                return 0;
            }
            if (g[i].size() == 1)
                head = i;
        }
        v1.push_back(head);
        int cur = g[head][0], nxt;
        rep(i, 1, n - 1)
        {
            if (i == n - 1)
                v1.push_back(cur);
            for (auto x : g[cur])
            {
                if (x != head)
                {
                    head = cur;
                    v1.push_back(cur);
                    cur = x;
                    break;
                }
            }
        }
        int a[6][3] = { {1,2,3},{1,3,2},{2,1,3},{2,3,1},{3,1,2},{3,2,1} };
        ll res1 = INF, res2;
        rep(i, 0, 5)
        {
            int cnt = 0;
            ll tmp1 = 0;
            rep(j, 0, n-1)
            {
                tmp1 += c[a[i][cnt++]-1][v1[j]];
                cnt %= 3;
            }
            if (tmp1 < res1)
            {
                res1 = tmp1;
                res2 = i;
            }
        }
        cout << res1 << endl;
        vector<int> res(n+1);
        rep(i, 0, n-1)
            res[v1[i]]=a[res2][i % 3];
        rep(i, 1, n)
            cout << res[i] << " ";
        return 0;
    }
  • 相关阅读:
    UVa 820 因特网带宽(最大流)
    UVa 1001 奶酪里的老鼠(Dijkstra或Floyd)
    UVa 821 网页跳跃(Floyd)
    UVa 11624 大火蔓延的迷宫
    UVa 10881 蚂蚁
    UVa 11300 分金币
    UVa 11729 突击战
    《额尔古纳河右岸》读书笔记
    HDU 1083 Courses(二分图匹配模板)
    UVa 10618 跳舞机
  • 原文地址:https://www.cnblogs.com/dealer/p/13277166.html
Copyright © 2011-2022 走看看