Codeforces 1242C Sum Balance（状压DP）

题目链接

题意

给你 $n$ 行每行若干整数（所有整数互不相同），每行拿出任意一个整数，再给每行放回一个拿出的整数，求这样操作一次后所有行的整数和相同的操作方案

思路

第一次打DIV1，没想到代码能力太弱想到思路却没敲出来，题还是刷太少。

先验证所有整数和除以n是否存在余数判断

之后对每个点求其所在的环，每个环长度不超过n，所以这里复杂度是 $O(n\ast 5000\ast n)$。
将每个环上以环包含那几行（二进制）为key，将元素下标存入vector。
之后求一个包含所有行的集合，其中各个环不能相交的方案。这个使用状压dp解决

最后根据 $dp[(1<<n)-1]$ 判断是否有解，和输出路径。

整体复杂度 $O(n\ast n\ast 5000+3^n)$

代码

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

#define ll long long
#define _p pair<ll,ll>
#define fi first
#define se second

ll n, sum, up[20], e[20][5005], rsum[20];
_p to[20][5005];
map<ll,_p> mp;
vector<_p> path[1<<15];
ll vis[1<<15], dp[1<<15], pre[1<<15];
ll book[20];
_p ans[20];

int main() {
    scanf("%lld",&n);
    for(ll i = 1; i <= n; ++i) {
        scanf("%lld",&up[i]);
        for(ll j = 0; j < up[i]; ++j) {
            scanf("%lld",&e[i][j]);
            sum += e[i][j];
            rsum[i] += e[i][j];
            mp[e[i][j]] = {i,j};
        }
    }
    if(sum%n) return !printf("No\n");
    // build
    for(ll i = 1; i <= n; ++i) {
        for(ll j = 0; j < up[i]; ++j) {
            to[i][j] = mp.count(sum/n-(rsum[i]-e[i][j])) ? mp[sum/n-(rsum[i]-e[i][j])] : make_pair(0ll,0ll);
        }
    }
    // getcircle
    for(ll i = 1; i <= n; ++i) {
        for(ll j = 0; j < up[i]; ++j) {
            _p now = {i,j};
            vector<_p> tmp;
            memset(book,0,sizeof(book));
            ll sta = 0, f = 0;
            while(1) {
                sta += (1<<(now.fi-1));
                book[now.fi] = 1;
                tmp.push_back(now);
                now = to[now.fi][now.se];
                if(now == make_pair(0ll,0ll)) break;
                if(book[now.fi]) {
                    if(now == make_pair(i,j)) f = 1;
                    break;
                }
            }
            if(f && path[sta].size() == 0) {
                path[sta] = tmp;
                vis[sta] = 1;
            }
        }
    }
    // getans
    dp[0] = 1;
    for(ll i = 1; i < (1<<n); ++i) {
        for(ll j = i; j; j = i&(j-1)) {
            if(vis[j] && dp[i^j]) {
                dp[i] = 1;
                pre[i] = j;
            }
        }
    }
    if(dp[(1<<n)-1] == 0) return !printf("No\n");
    printf("Yes\n");
    ll now = (1<<n)-1;
    while(now) {
        _p la;
        ll f = 0;
//        printf("%lld -\n",pre[now]);
        for(auto v : path[pre[now]]) {
//            printf("%lld -- %lld\n",v.fi,v.se);
            if(f) ans[v.fi] = make_pair(e[v.fi][v.se],la.fi);
            ++f;
            la = v;
        }
        ans[path[pre[now]][0].fi] = make_pair(e[path[pre[now]][0].fi][path[pre[now]][0].se], la.fi);
        now ^= pre[now];
    }
    for(ll i = 1; i <= n; ++i) printf("%lld %lld\n",ans[i].fi,ans[i].se);
    return 0;
}