牛客算法周周练5 (A 状压+容斥 B 数学推导 D 分层图 E bitset优化01背包

题目链接

A-多彩的树

做法参考来自：博客

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const ll mod=1e9+7;
const int N=5e4+10,M=1e3+100;
int col[N], p[55], vis[N];
ll sum, ans[15], f[15] = {1}, dp[M], num[M], dp2[M];
vector<int> G[N];
void dfs(int x)
{
	int i, v;
	vis[x] = 1, sum++;
	for(i=0;i<G[x].size();i++)
	{
		v = G[x][i];
		if(vis[v] || p[col[v]]==0)
			continue;
		dfs(v);
	}
}
int main(void)
{
	int n, k;
	scanf("%d%d", &n, &k);
	f[0]=1;
	for(int i=1;i<=k;i++) f[i] = f[i-1]*131%mod;
	for(int i=1;i<=n;i++) scanf("%d", &col[i]);
	for(int i=1;i<=n-1;i++){
        int x,y;
		scanf("%d%d", &x, &y);
		G[x].push_back(y);
		G[y].push_back(x);
	}


	for(int i=1;i<(1<<k);i++)
	{
		int x = 0;
		for(int j=0;j<=k-1;j++)
		{
			p[j+1] = 0;
			if(i&(1<<j)) p[j+1] = 1, x++;
		}
		for(int j=1;j<=n;j++) vis[j] = 0;
		for(int j=1;j<=n;j++)
		{
			if(vis[j]==0 && p[col[j]])
			{
				sum = 0;
				dfs(j);
				dp2[i] = dp[i] = (dp[i]+sum*(sum-1)/2)%mod;
				num[i] = x;
			}
		}
	}
	for(int i=1;i<(1<<k);i++)
	{
		for(int j=1;j<(1<<k);j++)
		{
			if((i|j)==i && i!=j)//子集
			{
				if((num[i]-num[j])%2)
					dp2[i] = (dp2[i]-dp[j]+mod)%mod;
				else
					dp2[i] = (dp2[i]+dp[j])%mod;
			}
		}
		ans[num[i]] = (ans[num[i]]+dp2[i])%mod;
	}
	ll res = 0;
	for(int i=1;i<=k;i++)
		res = (res+ans[i]*f[i])%mod;
	res = (res+n*131)%mod;
	printf("%lld\n", res);
	return 0;
}

B-求幂

这题我推了好几种公式，发现小范围是对的，大范围总是不对，做法参考来自：简书

做法很妙，学到了。。

#include <bits/stdc++.h>
using namespace std;
const int mod=1e9+7;

int main() {
    int n;
    scanf("%d",&n);
    set<int> S;
    int res=1LL*n*(n*2-1)%mod;//n*n 是底数为1的 n*(n-1)是i^j [i(2->n) j(1->n)]
    for(int i=2;i<=n; ++i) {
        if(S.find(i)!=S.end()) continue; // 如果已经存在则跳过
        long long temp=i;
        int cnt=0;
        while(temp<=n) { // 求小于n的最大幂
            S.insert(temp);
            temp=temp*i;
            cnt++;
        }
        for(int x=1; x<=cnt; ++x){ // 遍历统计到的幂 1/2开始 1/1已经提前处理:n*(n-1)
            for(int y=x+1; y<=cnt; ++y){
                res=(res+n/(y/__gcd(x,y))*2LL)%mod;
            }
        }
    }
    printf("%d\n",res);
    return 0;
}

D-小雨坐地铁

这题我之前写过题解的了：博客

E-简单瞎搞题

这题本质是一个01背包，dp[i][j]代表前i个数  能否到达j值。最后统计dp[n]中为1的数量。

数组方式：100*100*1000000  超时。用bitset优化：biset优化01背包学习博客

#include<cstdio>
#include<iostream>
#include<cstdlib>
#include<cstring>
#include<algorithm>
#include<queue>
#include<vector>
#include<ctime>
#include<cmath>
#include<bitset>
using namespace std;
typedef long long ll;
const int N =1e6+10;
bitset<N>f[110];
 
 
int a[N],l[N],r[N];
int main() {
 
    int n;
    scanf("%d",&n);
    for(int i = 1;i <= n;++i) cin>>l[i]>>r[i];
 
 
 
 
    f[0][0] = 1;
    //f[0]|=f[0]<<1;
    //cout<<f[0]<<endl;
    for(int i = 1;i <= n;++i) {
        //f[i] = f[i - 1];
        for(int x=l[i];x<=r[i];++x){
            f[i]|=f[i-1]<<(x*x);
        }
        //cout<<f[i]<<endl;
    }
 
    printf("%d\n",f[n].count());
}