Submission #1359197

---

Source Code Expand

#include<iostream>
#include<vector>
#include<algorithm>
#include<cstring>
#include<cstdio>
#include<cmath>
#include<cstdlib>
#include<ctime>
#include<queue>
#include<set>
using namespace std;
typedef long long LL;
int gi() {
	int w=0;bool q=1;char c=getchar();
	while ((c<'0'||c>'9') && c!='-') c=getchar();
	if (c=='-') q=0,c=getchar();
	while (c>='0'&&c <= '9') w=w*10+c-'0',c=getchar();
	return q? w:-w;
}
const int mod=1e9+7;
struct M{ int a[3][3]; };
inline M operator * (const M &a,const M &b) {
	M ans={};
	for (int i=0;i<3;i++)
		for (int j=0;j<3;j++)
			for (int k=0;k<3;k++)
				ans.a[i][j]=(ans.a[i][j]+1LL*a.a[i][k]*b.a[k][j])%mod;
	return ans;
}
const M trans=(M){
	{
		{1,2,1},//0->0 不放 0->1 两个球挑一个放 0->2 两个一起放
		{0,1,1},//1->0 不合法 1->1 不放 1->2 放剩下的一个
		{1,2,2}//2->0 新建 2->1 新建并选一个放 2->2 不放或新建并放两个
	}
};
const M O=(M){
	{
		{1,2,1},
		{0,1,1},
		{0,0,1}
	}
};
int main()
{
	int n=gi(),m=gi(),i,x=-1,last,y;
	M a={},b;a.a[0][0]=1;
	for (i=1;i<=m;i++) {
		last=x;x=gi();
		y=x-last-1;b=trans;
		while (y) {
			if (y&1) a=a*b;
			b=b*b;y>>=1;
		}
		if (x!=n) a=a*O;
	}
	if (x!=n) {
		y=n-x-1;b=trans;
		while (y) {
			if (y&1) a=a*b;
			b=b*b;y>>=1;
		}
	}
	printf("%d\n",a.a[0][2]);
	return 0;
}

Submission Info

Judge Result