zoukankan      html  css  js  c++  java
  • 腾讯笔试——彩色楼房

    # 彩色楼房

    [Time Limit: 5000 ms quad Memory Limit: 262144 kB ]

    题目

    小Q给你n个数1,2,3...n,代表n个楼,第i个楼的高度为i,每个楼都有一个颜色
    现在将这些楼随机排列在一条直线上,问有多少种排列方式,满足从最左边往右看(站在左边很远的地方看)能看到L-1次颜色的变变化,答案对1e9+9取模
    如果两个相同颜色楼的高度分别为H1,H2(H1<H2),H1在左边,且H1 H2之间的楼都比H1矮,那么站在左边来看就是一种颜色
    你能看到一个楼的前提是这个楼之前的楼都比它矮

    input

    第一行输入两个整数n,L(1<=n<=1296),(1<=L<=n)
    第二行输入n个整数Ci表示每个楼的颜色(1<=Ci<=n)

    output

    输出一个整数,代表方案数,并对1e9+9取模

    思路

    以下做法参考厦大一位大佬...

    step1

    (O(N^3)) 暴力 (dp),从后往前 (dp)
    (dp[i][k][j]) 表示 到第 (i) 个楼房时,可以看到 (k) 个颜色,并且最左边楼层颜色为 (j)
    对于 (dp[i][k][j]) 存在以下转移:

    1. 首先考虑最左边的楼层是 (i) 后面的楼层,把 (i) 楼层插到后面去,那么有

    [dp[i][k][j] = dp[i+1[k][j]*(n-i) ]

    1. 考虑最左边的楼层是 (i) 楼层,也就是当 (j==a[i]) 时,有

    [dp[i][k][j] += dp[i+1][k][j] \ dp[i][k][j] += dp[i+1][k-1][sum_{j'=1}^{j' eq j}j'] ]

    因为存在 (j' eq j) 的判断,在 (for) 一遍是很浪费时间的,所以可以开一个数组 (sum)(sum[i][k]) 表示 (sum_{j=1}^{j=n}dp[i][k][j]),然后用减法来得到答案。

    时间复杂度 (O(N^3))、空间复杂度(O(N^3))
    代码:(这代码只能跑 (n)(500) 以内)

    void solve1() {
    	mes(sum, 0);
    	dp[n][1][a[n]] = 1;
    	sum[n][1] = 1;
    	for(int i=n-1; i>=1; i--) {
    		for(int k=1; k<=m; k++) {
    			for(int j=1; j<=n; j++) {
    				if(j == a[i]) {
    					dp[i][k][j] = (dp[i+1][k][j]*(n-i+1)%mod+sum[i+1][k-1]-dp[i+1][k-1][j])%mod+mod;
    					dp[i][k][j] %= mod;
    				} else {
    					dp[i][k][j] = dp[i+1][k][j]*(n-i)%mod;
    				}
    				sum[i][k] = (sum[i][k] + dp[i][k][j])%mod;
    			}
    		}
    	}
    	printf("%lld
    ", sum[1][m]);
    }
    

    step2

    通过观察式子不难发现,由于 (k) 只和 (k-1) 的答案有关,只要从大到小枚举 (k),第一维是可以滚动优化掉。
    观察代码,我们发现第三层循环中,只有当 (j == a[i]) 时,转移比较复杂,对于其他的 (j),转移过程完全一样。所以我们可以简化这层循环。
    首先把这层循环里的式子尽量提取出来。
    对于 (dp) 式子,可以直接计算 (j==a[i]) 的情况,把 (j eq a[i]) 的情况留在 (for) 中。这一步很容易做到。
    对于 (sum) 式子,可以发现(j eq a[i])部分,(dp) 整体乘 (n-i),所以这一部分可以写成 (sum[k]*(n-i)-dp[k][j]*(n-i)),然后在加上特别计算出的 (j==a[i])的情况的 (dp) 值就可以了。
    时间复杂度(O(N^3)),空间复杂度(O(N^2))

    void solve2() {
    	mes(sum, 0);
    	dp[1][a[n]] = 1;
    	sum[1] = 1;
    	for(int i=n-1; i>=1; i--) {
    		for(int k=m; k>=1; k--) {
    			{
    				int j = a[i];
    				sum[k] = ((sum[k]*(n-i)%mod - dp[k][j]*(n-i)%mod)%mod+mod)%mod;
    				dp[k][j] = (dp[k][j]*(n-i+1)%mod+sum[k-1]-dp[k-1][j])%mod+mod;
    				dp[k][j] %= mod;
    				sum[k] = (sum[k] + dp[k][j])%mod;
    			}
    			for(int j=1; j<=n; j++) {
    				if(j != a[i]) {
    					dp[k][j] = dp[k][j]*(n-i)%mod;
    				}
    				// printf("dp[%d][%d] = %lld
    ", k, j, dp[k][j]);
    			}
    		}
    	}
    	printf("%lld
    ", sum[m]);
    }
    

    step3

    还是在第三层循环上想办法,我们看到这一层除了(j==a[i]) 外都乘上了一个 ((n-i)),那么如果我们让 (j==a[i]) 时也乘上一个 ((n-i)),就可以统一整维的 (dp) 了。那么我们在保存的时候可以保存 (frac{dp[k][j]}{(n-i)}) 的值,运算的时候在把这个原本应有的值乘回来就可以了。
    由于每一个 (i) 都要乘上一个 (frac{1}{n-i}),也就是说在对于第 (i) 个楼房而言,一共乘上了 (frac{1}{prod_{x=1}^{x} n-x})
    这一步也可以做到,只要每次在运算 (j==a[i]) 时,让 (dp[k][j]) 乘上他应该缩小的倍数就可以了。
    这样处理后,时间复杂度 (O(N^2)),空间复杂度 (O(N^2))

    /*************************************************************** 
        > File Name    : a.cpp
        > Author       : Jiaaaaaaaqi
        > Created Time : Tue 10 Sep 2019 10:21:23 PM CST
     ***************************************************************/
    
    #include <map>
    #include <set>
    #include <list>
    #include <ctime>
    #include <cmath>
    #include <stack>
    #include <queue>
    #include <cfloat>
    #include <string>
    #include <vector>
    #include <cstdio>
    #include <bitset>
    #include <cstdlib>
    #include <cstring>
    #include <iostream>
    #include <algorithm>
    #include <unordered_map>
    #define  lowbit(x)  x & (-x)
    #define  mes(a, b)  memset(a, b, sizeof a)
    #define  fi         first
    #define  se         second
    #define  pb         push_back
    #define  pii        pair<int, int>
    
    typedef unsigned long long int ull;
    typedef long long int ll;
    const int    maxn = 2e3 + 10;
    const int    maxm = 1e5 + 10;
    const ll     mod  = 1e9 + 9;
    const ll     INF  = 1e18 + 100;
    const int    inf  = 0x3f3f3f3f;
    const double pi   = acos(-1.0);
    const double eps  = 1e-8;
    using namespace std;
    
    ll n, m;
    int cas, tol, T;
    
    ll sum[maxn];
    ll dp[maxn][maxn];
    ll a[maxn];
    
    ll fpow(ll a, ll b) {
    	ll ans = 1;
    	while(b) {
    		if(b&1)	ans = ans*a%mod;
    		a = a*a%mod;
    		b >>= 1;
    	}
    	return ans;
    }
    
    int main() {
    	// freopen("in", "r", stdin);
    	scanf("%lld%lld", &n, &m);
    	for(int i=1; i<=n; i++) {
    		scanf("%lld", &a[i]);
    	}
    	mes(sum, 0);
    	dp[1][a[n]] = 1;
    	sum[1] = 1;
    	ll base = 1;
    	for(int i=n-1; i>=1; i--) {
    		for(int k=m; k>=1; k--) {
    			{
    				int j = a[i];
    				sum[k] = ((sum[k]*(n-i)%mod - dp[k][j]*(n-i)%mod*base%mod)%mod+mod)%mod;
    				dp[k][j] = ((dp[k][j]*base%mod*(n-i+1)%mod+sum[k-1])%mod-dp[k-1][j]*base%mod)%mod+mod;
    				dp[k][j] %= mod;
    				sum[k] = (sum[k] + dp[k][j])%mod;
    				dp[k][j] = dp[k][j] * fpow(base*(n-i)%mod, mod-2)%mod;
    				// printf("dp[%d][%d] = %lld
    ", k, j, dp[k][j]*base);
    			}
    		}
    		base = base*(n-i)%mod;
    	}
    	printf("%lld
    ", sum[m]%mod);
    	return 0;
    }
    
  • 相关阅读:
    ES中对应的SQL的count(distinct 列名) java实现
    maven使用
    自旋锁
    Java手写死锁并用jps和jstack查看
    已知二叉树前序和中序,算法写出后续遍历的结果
    Idea里搭建SpringMVC项目,部署的包下没有lib
    Spring配置文件中关于bean标签的id属性和name属性的说明
    ORA-12519: TNS:no appropriate service handler found 解决
    springmvc常用注解标签详解
    Spring/SpringMvc 配置文件常用标签解释
  • 原文地址:https://www.cnblogs.com/Jiaaaaaaaqi/p/11504103.html
Copyright © 2011-2022 走看看