Codeforces Round #599 (Div. 2)

题目链接：https://codeforces.com/contest/1243

C. Tile Painting

题意：给定一个长度为n的布条，要求间隔差为n的因子的位置对应的颜色要相同，求最多能染多少种颜色。
题解：本质就是找质因子公约数，傻逼题，但是窝为啥wa了好几发，，

#include<bits/stdc++.h>
using namespace std;
#define ll long long
int main()
{
    ll n;
    scanf("%I64d",&n);
    ll h=n;
    ll m=sqrt(n);
    ll ans=n;
    for(ll i=2;i*i<=h;i++){
        if(h%i==0){
            if(ans!=n){//two factors
                ans=1;break;
            }
            ans=i;
            while(h%i==0) h/=i;
        }
    }
    if(h!=1&&h!=n) ans=1;
    printf("%I64d\n",ans);
    return 0;
}

D. 0-1 MST

最小生成树 set
给定n个点，m个点对，n,m<=1e5
m个点之间点对值为1，其余点对值为0，求最小生成树
题解：如果模拟着来，那么有n*(n-1)/2个边，直接爆。
考虑用set存储当前还没被用过的点；用set存和u相连边权为1的点；每次dfs搜索不在当前点相连连的点；最后计数总共跑多少次dfs，就对应跑了多少个超级结点，答案即为超级结点数-1。

//By zhy123,
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
int n,m;
set<int> g[100010];
set<int> s;
void dfs(int x){
	s.erase(x);
	set<int>::iterator it;
	for(it=s.begin();it!=s.end();){
		int ver=*it;
		if(!g[x].count(ver)){
			dfs(ver);
		}
		it=s.upper_bound(ver);
	}
}
int main()
{
	scanf("%d%d",&n,&m);
	for(int i=1;i<=n;i++){
		s.insert(i);
	}
	for(int i=0;i<m;i++){
		int a,b;
		scanf("%d%d",&a,&b);
		g[a].insert(b);
		g[b].insert(a);
	}
	int res=0;
	//printf("%d\n",*s.begin());
	while(!s.empty()){
		res++;
		//printf("%d\n",*s.begin());
		dfs(*s.begin());
	}
	printf("%d\n",res-1);
  return 0;
}

E-Sum Balance（状压dp 图论）

参考：https://www.cnblogs.com/qscqesze/p/11813445.html
题意：k组数，每组ni个，数互不相同
把每组数中的一个移到其他组（或者不移动），使得最终每组数的个数不变且总和相等
k<=15,ni<=5000
题解：原来每个数互不相同，k最大15这些，都是为了状压dp做准备的orz，预处理找环，用状压转移dp

#include <bits/stdc++.h>
using namespace std;
const int maxn=16;
#define ll long long
vector<pair<pair<int,ll>,int> > ans[1<<maxn];

map<ll,int> color;
vector<int> a[maxn];
ll sum[maxn];
int k,dp[1<<maxn];

int main(){
    scanf("%d",&k);
    ll all_sum=0;
    for(int i=0;i<k;i++){
        int n;scanf("%d",&n);
        for(int j=0;j<n;j++){
            ll x;scanf("%I64d",&x);
            a[i].push_back(x);
            all_sum+=x;
            sum[i]+=x;
            color[a[i][j]]=i;
        }
    }
    if(all_sum%k!=0){
        puts("NO");
        return 0;
    }
    all_sum/=k;
    for(int i=0;i<k;i++){
        for(int j=0;j<a[i].size();j++){
            ll cur=a[i][j];
            int used=0;
            bool ok=1;
            vector<pair<pair<int,ll>,int> > an;
            do{
                int c1=0;
                auto it=color.find(cur);
                if(it!=color.end())
                    c1=it->second;
                else{
                	ok=0;break;
				}
                if(used&(1<<c1)){
                    ok=0;break;
                }
                used|=(1<<c1);
                cur=cur+(all_sum-sum[c1]);
                auto c12=color.find(cur);
                if(c12!=color.end()){
                    an.push_back({{c12->second,cur},c1});
                }
            }while(cur!=a[i][j]);
            if(ok){
                dp[used]=1;
                ans[used]=an;//
            }
        }
        for(int i=0;i<(1<<k);i++){
            if(dp[i]) continue;
            for(int j=i;j>0;j=(j-1)&i){
                if(dp[j]&&dp[i&(~j)]){
                    dp[i]=1;
                    ans[i]=ans[j];
                    for(auto &x:ans[i&(~j)])
                        ans[i].push_back(x);
                    break;
                }
            }
        }
        
    }
    int x=(1<<k)-1;
    if(dp[x]){
        puts("Yes");
        sort(ans[x].begin(),ans[x].end());
        for(auto a:ans[x]){
            printf("%I64d %d\n",a.first.second,a.second+1);
        }
    }else{
        puts("No");
    }
    return 0;
}