結果

問題 No.650 行列木クエリ
ユーザー ryoissyryoissy
提出日時 2018-02-09 23:48:05
言語 C++11
(gcc 11.4.0)
結果
WA  
実行時間 -
コード長 3,711 bytes
コンパイル時間 2,050 ms
コンパイル使用メモリ 184,552 KB
実行使用メモリ 116,152 KB
最終ジャッジ日時 2024-04-17 10:39:28
合計ジャッジ時間 5,159 ms
ジャッジサーバーID
(参考情報)
judge1 / judge4
このコードへのチャレンジ
(要ログイン)

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 WA -
testcase_01 WA -
testcase_02 WA -
testcase_03 RE -
testcase_04 WA -
testcase_05 WA -
testcase_06 WA -
testcase_07 AC 34 ms
30,336 KB
testcase_08 WA -
testcase_09 WA -
testcase_10 WA -
権限があれば一括ダウンロードができます
コンパイルメッセージ
main.cpp: In function ‘int main()’:
main.cpp:186:35: warning: format ‘%d’ expects argument of type ‘int*’, but argument 3 has type ‘__gnu_cxx::__alloc_traits<std::allocator<long long int>, long long int>::value_type*’ {aka ‘long long int*’} [-Wformat=]
  186 |                         scanf("%d%d%d%d%d%*c",&v,&C[0][0],&C[0][1],&C[1][0],&C[1][1]);
      |                                  ~^
      |                                   |
      |                                   int*
      |                                  %lld
main.cpp:186:37: warning: format ‘%d’ expects argument of type ‘int*’, but argument 4 has type ‘__gnu_cxx::__alloc_traits<std::allocator<long long int>, long long int>::value_type*’ {aka ‘long long int*’} [-Wformat=]
  186 |                         scanf("%d%d%d%d%d%*c",&v,&C[0][0],&C[0][1],&C[1][0],&C[1][1]);
      |                                    ~^
      |                                     |
      |                                     int*
      |                                    %lld
main.cpp:186:39: warning: format ‘%d’ expects argument of type ‘int*’, but argument 5 has type ‘__gnu_cxx::__alloc_traits<std::allocator<long long int>, long long int>::value_type*’ {aka ‘long long int*’} [-Wformat=]
  186 |                         scanf("%d%d%d%d%d%*c",&v,&C[0][0],&C[0][1],&C[1][0],&C[1][1]);
      |                                      ~^
      |                                       |
      |                                       int*
      |                                      %lld
main.cpp:186:41: warning: format ‘%d’ expects argument of type ‘int*’, but argument 6 has type ‘__gnu_cxx::__alloc_traits<std::allocator<long long int>, long long int>::value_type*’ {aka ‘long long int*’} [-Wformat=]
  186 |                         scanf("%d%d%d%d%d%*c",&v,&C[0][0],&C[0][1],&C[1][0],&C[1][1]);
      |                                        ~^
      |                        

ソースコード

diff #

#include <bits/stdc++.h>
#define MOD 1000000007LL
using namespace std;
typedef long long ll;
typedef pair<int,int> P;

ll extgcd(ll a,ll b,ll& x,ll& y){
	ll d=a;
	if(b!=0LL){
		d=extgcd(b,a%b,y,x);
		y-=(a/b)*x;
	}else{
		x=1;
		y=0;
	}
	return d;
}

ll mod_inverse(ll a,ll m){
	ll x,y;
	extgcd(a,m,x,y);
	return (m+x%m)%m;
}

typedef vector<ll> vec;
typedef vector<vec> mat;

mat mul(mat A,mat B){
	mat C(A.size(),vec(B[0].size()));
	for(int i=0;i<A.size();i++){
		for(int k=0;k<B.size();k++){
			for(int j=0;j<B[0].size();j++){
				C[i][j]=(C[i][j]+(A[i][k]*B[k][j]%MOD))%MOD;
			}
		}
	}
	return C;
}
int N;

class segtree{
public:
	vector<mat> dp;
	mat ta;
	segtree(){
		ta.push_back(vec());
		ta.push_back(vec());
		ta[0].resize(2);
		ta[1].resize(2);
		ta[0][0]=1;
		ta[0][1]=0;
		ta[1][0]=0;
		ta[1][1]=1;
	}
	void init(int n){
		dp.clear();
		N=1;
		while(N<n){
			N*=2;
		}
		for(int i=0;i<N*2;i++){
			dp.push_back(mat());
			dp[i].push_back(vec());
			dp[i].push_back(vec());
			dp[i][0].resize(2);
			dp[i][1].resize(2);
			dp[i][0][0]=1;
			dp[i][0][1]=0;
			dp[i][1][0]=0;
			dp[i][1][1]=1;
		}
	}
	void update(int k,mat v){
		k+=N-1;
		dp[k]=v;
		while(k>0){
			k=(k-1)/2;
			dp[k]=mul(dp[k*2+1],dp[k*2+2]);
		}
	}

	mat query(int a,int b,int k=0,int l=0,int r=N){
		if(b<=l || r<=a)return ta;
		if(a<=l && r<=b)return dp[k];
		int mid=(l+r)/2;
		mat vl=query(a,b,k*2+1,l,mid);
		mat vr=query(a,b,k*2+2,mid,r);
		return mul(vl,vr);
	}
};

int n,q;

mat val[100001];
mat sum[100001];
vector<int> G[100001];
int depth[100001];
int siz[100001];
int hlsize=0;
vector<int> hls[100001];
int which[100001][2];
int parent[100001];
segtree seg[100001];

int dfs(int v,int p,int c){
	depth[v]=c;
	sum[v]=mul(sum[v],val[v]);
	int cnt=1;
	for(int i=0;i<G[v].size();i++){
		if(G[v][i]!=p){
			cnt+=dfs(G[v][i],v,c+1);
		}
	}
	return (siz[v]=cnt);
}

void constr(int v,int p,int id){
	which[v][0]=id;
	which[v][1]=hls[id].size();
	hls[id].push_back(v);

	int b=-1;
	for(int i=0;i<G[v].size();i++){
		if(G[v][i]!=p){
			if(b==-1){
				b=G[v][i];
			}else if(siz[G[v][i]]>siz[b]){
				b=G[v][i];
			}
		}
	}
	for(int i=0;i<G[v].size();i++){
		if(G[v][i]!=p){
			if(b==G[v][i]){
				constr(G[v][i],v,id);
			}else{
				parent[hlsize]=v;
				constr(G[v][i],v,hlsize++);
			}
		}
	}
}

int main(void){
	//printf("done\n");
	scanf("%d",&n);
	for(int i=0;i<n;i++){
		val[i].push_back(vec());
		val[i].push_back(vec());
		val[i][0].resize(2);
		val[i][1].resize(2);
		val[i][0][0]=1;
		val[i][0][1]=0;
		val[i][1][0]=0;
		val[i][1][1]=1;
		sum[i].push_back(vec());
		sum[i].push_back(vec());
		sum[i][0].resize(2);
		sum[i][1].resize(2);
		sum[i][0][0]=1;
		sum[i][0][1]=0;
		sum[i][1][0]=0;
		sum[i][1][1]=1;
	}
	for(int i=0;i<n-1;i++){
		int a,b;
		scanf("%d%d%*c",&a,&b);
		G[a].push_back(b);
		G[b].push_back(a);
	}
	scanf("%d%*c",&q);
	dfs(0,-1,0);
	hlsize++;
	constr(0,-1,0);
	//printf("done\n");
	for(int i=0;i<hlsize;i++){
		seg[i].init(hls[i].size());
	}
	
	for(int i=0;i<q;i++){
		char t;
		scanf("%c",&t);
		if(t=='x'){
			mat C(2,vec(2));
			int v;
			scanf("%d%d%d%d%d%*c",&v,&C[0][0],&C[0][1],&C[1][0],&C[1][1]);
			seg[which[v][0]].update(which[v][1],C);
		}else{
			int f,t;
			scanf("%d%d%*c",&f,&t);
			mat ans(2,vec(2));
			ans[0][0]=1;
			ans[0][1]=0;
			ans[1][0]=0;
			ans[1][1]=1;
			while(which[f][0]!=which[t][0]){
				ans=mul(seg[which[t][0]].query(0,which[t][1]+1),ans);
				t=parent[which[t][0]];
				//printf("%d ",t);
				//printf("%lld %lld %lld %lld\n",ans[0][0],ans[0][1],ans[1][0],ans[1][1]);
			}
			//sprintf("end\n");
			ans=mul(seg[which[f][0]].query(which[f][1],which[t][1]+1),ans);
			printf("%lld %lld %lld %lld\n",ans[0][0],ans[0][1],ans[1][0],ans[1][1]);
		}

	}
	return 0;
}
0